ML.NET 展示哪个分数对应哪个标签

我在使用ML.Net进行文本解释的分类器。预测结果有一个float[]类型的分数列和一个预测的标签。这种方法有效,因为最高分对应于预测的标签,但其他分数只是没有特定顺序的浮点数。我如何知道哪个分数对应哪个标签?我如何查看第二高权重的标签是什么?

例如,我得到这样的结果:0.000050090.008930760.12747630.62097870.2425644

0.6是我的预测标签,但我还需要查看0.24对应的标签,以便了解它为什么会混淆。

标签是诸如“问候”或“笑话”这样的文本字符串,这些在管道中被字典化了,所以这可能是它们没有按正确顺序排列的原因?

在ML.Net中,有没有办法将两者联系起来?展示哪个分数对应哪个标签?


回答:

您可以使用以下代码获取与分数对应的标签:

string[] scoreLabels;model.TryGetScoreLabelNames(out scoreLabels);

更多详细信息可以在这里找到 这里这里

请注意,这可能会随着即将推出的ML.NET 0.6 API而改变。这些API将直接暴露Schema,并启用获取此信息(以及其他有用信息)。这可能类似于今天的TryGetScoreLabelNames的工作方式。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注