如何解决ONNX C#:我如何阅读这个对象并提取概率值?
我已经保存了一个 ONNX 转换的预训练 RFC 模型,我正尝试在我的 API 中使用它。我可以调用保存的模型并进行预测,但是我无法获得预测值。反应很复杂,我似乎无法弄清楚。 这是调试模式下的预测值形状:
这里是在 python 中,我正在尝试获取第二个值:
这是我迄今为止尝试过的,但我只得到字符串值。
var result = _inferencesession.Run(new List<NamedOnnxValue>
{
NamedOnnxValue.CreateFromTensor("input",attritionData.AsTensor())
});
var scoreValue = result.LastOrDefault().Value;
var score = scoreValue.GetType()?.GetProperty("Value")?.Attributes;
解决方法
我找到了! 我必须使用 AsEnumerable 然后访问我想要的值:
var seq = result.LastOrDefault().AsEnumerable<NamedOnnxValue>();
var map = seq.First().AsDictionary<Int64,float>();
var values = map.First().Value; //this is what I want
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。