我有许多类和相应的特征向量,当我运行predict_proba()时,我会得到以下信息:
classes = ['one','two','three','one','three'] feature = [[0,1,1,0],[0,1,0,1],[1,1,0,0],[0,0,0,0],[0,1,1,1]] from sklearn.naive_bayes import BernoulliNB clf = BernoulliNB() clf.fit(feature,classes) clf.predict_proba([0,1,1,0]) >> array([[ 0.48247836, 0.40709111, 0.11043053]])
这是否意味着我已经通过训练示例将相应的索引分配给班级的第一次遇到,还是有这样的命令?
clf.getClasses() = ['one','two','three']?
只需使用.classes_分类器的属性即可恢复映射。在您的示例中给出:
.classes_
>>> clf.classes_ array(['one', 'three', 'two'], dtype='|S5')
还要感谢您在问题中添加了一个简约的复制脚本,只需复制并粘贴到IPython shell中,它就使回答变得非常容易:)