Пример #1
0
            public async Task <OnnxModelOutput> EvaluateAsync(OnnxModelInput input)
            {
                OnnxModelOutput             output  = new OnnxModelOutput();
                LearningModelBindingPreview binding = new LearningModelBindingPreview(learningModel);

                binding.Bind("data", input.data);
                binding.Bind("classLabel", output.classLabel);
                binding.Bind("loss", output.loss);
                LearningModelEvaluationResultPreview evalResult = await learningModel.EvaluateAsync(binding, string.Empty);

                return(output);
            }
Пример #2
0
            public async Task <OnnxModelOutput> EvaluateAsync(OnnxModelInput input)
            {
                binding.Bind("data", input.data);
                var result = await session.EvaluateAsync(binding, string.Empty);

                OnnxModelOutput output = new OnnxModelOutput();

                output.classLabel = (result.Outputs["classLabel"] as TensorString).GetAsVectorView();
                var predictions = result.Outputs["loss"] as IList <IDictionary <string, float> >;

                output.loss = predictions[0];

                return(output);
            }