Wie verwende ich die Binärklassifikation von Naive Bayes in CoreML?

Question

Ich versuche ein Zahlenklassifikationsmodell in CoreML zu erstellen und möchte den naiven Bayes Classifier verwenden, aber nicht finden, wie man es benutzt. Mein Algorithmus verwendet Naive Bayes

Answer 1

Im Moment coremltools Unterstützung nach nur Arten von Klassifikatoren:

• SVMs (Scikit-Learn)
• Neuronale Netze (Keras, Caffe)
• Entscheidungsbäume und ihre Ensembles (Scikit-Learn, xgboost)
• Lineare und logistische Regression (Scikit-Learn)

jedoch i, naive Bayes in Swift selbst durchzuführen ist nicht so schwer, überprüfen Sie diese implementation, for example.