如何在 CoreML 中使用朴素贝叶斯二元分类?

How to use Naive Bayes Binary Classification in CoreML?

我正在尝试在 CoreML 中构建数字分类模型,并想使用朴素贝叶斯分类器,但找不到如何使用它。我的算法使用的是朴素贝叶斯

目前,coremltools 仅支持以下类型的分类器:

  • 支持向量机 (scikitlearn)
  • 神经网络(Keras、Caffe)
  • 决策树及其集合(scikitlearn、xgboost)
  • 线性和逻辑回归 (scikitlearn)

但是,自己在 Swift 中实施朴素贝叶斯并不难,请查看此 implementation, for example