Scikit 学习朴素贝叶斯

Posted 2023-03-12

【中文标题】Scikit 学习朴素贝叶斯

【英文标题】：Scikit Learn Naive Bayes

【发布时间】：2016-05-01 02:17:54

【问题描述】：

我是 scikit learn 的新手，我对这个程序试图预测的内容感到困惑。

import numpy as np
X = np.array([[-1, -1], 
          [-2, -1], 
          [-3, -2], 
          [1, 1], 
          [2, 1], 
          [3, 2]])
Y = np.array([1, 1, 1, 2, 2, 2])
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()
clf.fit(X, Y)

print(clf.predict([[-0.8, -1]]))

如果我运行这个程序，我会得到：

[1]

据我所知，“X”是训练数据，我不确定“Y”是什么。如果我改变：

([[-0.8, -1]])

到

([[-0.8, 1]])

我明白了

[2]

我只需要定义一点。

【参考方案1】：

Y 是一个训练标签。 函数predict 返回预测标签。