python - sk learn中的朴素贝叶斯函数无法识别

Question

使用以下代码，我尝试加载数据集并对其执行 NB 算法。

from sklearn.naive_bayes import MultinominalNB 
import pandas as pd
import numpy as np

data = pd.read_csv('spambase.data.txt').as_matrix()
np.random.shuffle(data)

x = data[:, :48]
y = data[:, -1]

Ytrain = x[:-100,]
Ytrain = y[:-100,]
Xtest = x[-100:,]
Ytest = y[-100:,]

model = MultinominalNB()
model.fit(X.train, Y.train)

print "Classification rate for NB: "

但是，当我尝试运行它时出现以下错误:

python sample.py

Traceback (most recent call last):
File "sample.py", line 1, in <module>
from sklearn.naive_bayes import MultinominalNB
ImportError: No module named sklearn.naive_bayes

对这里出了什么问题有什么想法吗？

打印数据

Answer 1

您有一个拼写错误，请尝试:

从 sklearn.naive_bayes 导入 MultinomialNB