【发布时间】:2019-04-13 05:40:57
【问题描述】:
我是一名新的 Python 用户,并且一直在使用 scikit-learn 模块运行朴素贝叶斯分类器模型。 scikit learn Naïve Bayes documentation page 上的以下示例代码是否正确?
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("Number of mislabeled points out of a total %d points : %d"
gnb.fit() 函数不应该改为:
y_pred = gnb.fit(iris.data.drop(columns=['target']), iris.target).predict(iris.data)
也就是说,需要从预测数据集中手动移除响应变量。当一位同事指出我从 scikit-learn 文档页面抄录的代码有误时,我的模型获得了不合理的高准确度指标。
【问题讨论】:
标签: python scikit-learn