【发布时间】:2017-06-30 22:18:00
【问题描述】:
我正在尝试运行 MultinomiaL Naive bayes 并收到以下错误。给出了样本训练数据。测试数据完全相同。
def main():
text_train, targets_train = read_data('train')
text_test, targets_test = read_data('test')
classifier1 = MultinomialNB()
classifier1.fit(text_train, targets_train)
prediction1 = classifier1.predict(text_test)
样本数据:
Train:
category, text
Family, I love you Mom
University, I hate this course
【问题讨论】:
-
您的
text_train和targets_train的形状是什么?我也不认为 MultinomialNB 默认会处理文本数据。您需要对其进行预处理。
标签: machine-learning scikit-learn naivebayes multinomial