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); }
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); }