如何在Swift 5中从MLModel(机器学习模型)获得预测的信心

猎人

即使我的代码在尝试执行时非常简单,但基本上我还是希望分类器模型在实际的Xcode应用程序(而非运动场)上返回其认为预测的准确性的准确性/置信度值

我使用了一个伪造的“狗品种”分类器,例如“文本分类器”:

码:

do {

  if try DogClassiferModel().prediction(text: "value").confidence >= 90 {

  print("We have a high enough accuracy that this is the name of a dog breed")

  }

} catch let error {

  print(error)
}

我知道在Xcode的操场上,您可以使用测试数据执行此操作:

码:

  let data = try MLDataTable(contentsOf: URL(fileURLWithPath: "Path"))

  let (trainingData, testingData) = data.randomSplit(by: 0.8, seed: 5)

  let testClassifier = try MLTextClassifier(trainingData: trainingData, textColumn: "text", labelColumn: "recognized")

  // Getting the testing evaluation.
  let evaluationMetrics = testClassifier.evaluation(on: testingData)
  let evaluationAccuracy = (1.0 - evaluationMetrics.classificationError) * 100

  // We can print the accuracy with print(evaluationAccuracy).

想到:也许CoreML不能像我不知道的那样工作?

马蒂斯·霍勒曼斯

当你写的时候

try DogClassiferModel().prediction(text: "value")

返回的是一个DogClassiferModelOutput对象。如果模型的输出名为confidence,则可以通过编写以下内容获取它:

if let output = try DogClassiferModel().prediction(text: "value") {
  print(output.confidence)
}

但是,Core ML中的分类器模型通常以特殊方式处理。他们可以返回得分最高的类别的标签,或者返回包含所有标签的概率的字典。

找出这对您的模型如何工作的最佳方法是,查看Xcode中的mlmodel文件,单击箭头转到自动生成的源代码文件,然后查找“ Output”类。这将具有一个或多个您可以访问的属性(如上例所示)。

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

如何在pyspark中获得模型的预测

来自分类Dev

如何在Swift中获得iAds广告

来自分类Dev

如何使用机器学习模型预测特征稍有不同的数据?

来自分类Dev

如何使用训练有素的机器学习模型预测未标记的测试数据?

来自分类Dev

如何在Swift 5中遍历字典?

来自分类Dev

机器学习模型如何更新?

来自分类Dev

如何在Azure机器学习服务的初始化过程中传递模型名称?

来自分类Dev

如何在机器学习中训练用于从图像中去除背景的模型

来自分类Dev

如何在机器学习模型中引入“统一”稀疏/稀疏编码?

来自分类Dev

如何从R中的线性模型获得1000个预测?

来自分类Dev

如何从这些预训练模型中获得预测?

来自分类Dev

如何在XCTest for Swift中导入模型

来自分类Dev

如何在Swift中创建圆形按钮?

来自分类Dev

如何在Swift中隐藏特定警告

来自分类Dev

如何在Swift中调用deinit

来自分类Dev

如何在Swift中应用多个变换

来自分类Dev

如何在Swift中测试变量的类?

来自分类Dev

如何在Swift中准确使用MMWormhole?

来自分类Dev

如何在Swift中实现“共享按钮”

来自分类Dev

如何在Swift中制作日历视图

来自分类Dev

如何在Swift中销毁单身人士

来自分类Dev

如何在Swift中访问UIColor的扩展?

来自分类Dev

如何在Swift中旋转UIAlertController

来自分类Dev

如何在Swift中调试内存警告?

来自分类Dev

如何在Swift中访问程序参数?

来自分类Dev

如何在全局Swift中创建uialertcontroller

来自分类Dev

如何在Swift中调用UIViewController的transitionFromViewController?

来自分类Dev

如何在Swift中获取CConstPointer?

来自分类Dev

如何在Swift中返回序列?

Related 相关文章

热门标签

归档