如何解决CreateML分类器:预测概率
我正在操场上的CreateML中创建一个二进制表格分类器(使用代码,而不是CreateML UI)。
创建模型后,我希望能够对其进行测试。目前,在测试集上运行.predictions方法时,我可以获得离散标签(0或1)。但是,我希望将预测作为概率而不是根据阈值0.5计算得出的离散标签。
我知道我可以从Playground导出模型,然后放入一个Swift文件中,在其中可以加载模型,调用.predictions(即CoreML的.predictions方法),然后它可以让我访问每个类别的实际概率以及四舍五入的值。但是,在尝试调整模型时,很难为每个模型做到这一点,更不用说我将不得不编写额外的代码来解析我的测试csv文件并将要素列转换为数组并通过模型运行它,等等。等等。在Playground中,通过MLDataTable运行它要容易得多。
请让我知道是否有一种方法可以访问CreateML中每个类的概率!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。