Jak zapisać wyszkolony klasyfikator Naive Bayes na dysk i użyć go do przewidywania danych?
Mam następujący przykładowy program ze strony scikit-learn:
from sklearn import datasets
iris = datasets.load_iris()
from sklearn.naive_bayes import GaussianNB
gnb = GaussianNB()
y_pred = gnb.fit(iris.data, iris.target).predict(iris.data)
print "Liczba błędnie oznakowanych punktów : %d" % (iris.target != y_pred).sum()
0
1 odpowiedź
Klasyfikatory to po prostu obiekty, które można zzucić jak każdy inny.
Moja sugestia:
import cPickle
# zapisz klasyfikator
with open('my_dumped_classifier.pkl', 'wb') as fid:
cPickle.dump(gnb, fid)
# załaduj od nowa
with open('my_dumped_classifier.pkl', 'rb') as fid:
gnb_loaded = cPickle.load(fid)
0