如何在CoreML中使用Naive Bayes二进制分类？

Question

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

Answer 1

目前，coremltools只支持以下类型的分类：

• 支持向量机（scikitlearn）
• 神经网络（Keras，来自Caffe）
• 决策树和他们合奏（scikitlearn，xgboost）
• 线性和对数回归（scikitlearn）

然而，我在Swift中练习NaïveBayes并不难，请检查implementation, for example。