CreateML分类器:预测概率

时间:2020-09-01 17:03:16

标签: swift xcode swift-playground coreml createml

我正在操场上的CreateML中创建一个二进制表格分类器(使用代码,而不是CreateML UI)。

创建模型后,我希望能够对其进行测试。目前,在测试集上运行.predictions方法时,我可以获得离散标签(0或1)。但是,我希望将预测作为概率而不是根据阈值0.5计算得出的离散标签。

我知道我可以从Playground导出模型,然后放入一个Swift文件中,在其中可以加载模型,调用.predictions(即CoreML的.predictions方法),然后它可以让我访问每个类别的实际概率以及四舍五入的值。但是,在尝试调整模型时,很难为每个模型做到这一点,更不用说我将不得不编写额外的代码来解析我的测试csv文件并将要素列转换为数组并通过模型运行它,等等。等等。在Playground中,通过MLDataTable运行它要容易得多。

请让我知道是否有一种方法可以访问CreateML中每个类的概率!

0 个答案:

没有答案