0
我试图运行MultinomiaL Naive bayes并收到以下错误。给出样本训练数据。测试数据完全相似。ValueError:发现具有不一致样本数的输入变量:[1,14048]
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会默认处理文本数据。你需要预处理它。 –