ios - 如何在CoreML中使用朴素贝叶斯二进制分类？

Question

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

Answer 1

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

SVM(scikitlearn)

神经网络(Keras，Caffe)

决策树及其合奏(scikitlearn，xgboost)

线性和逻辑回归(scikitlearn)

但是，亲自在Swift中实现朴素贝叶斯并不难，请检查此 implementation, for example。