我正在尝试重现看到的教程 here .
在我将 .fit
方法添加到我的训练集之前,一切都很完美。
这是我的代码示例:
# TRAINING PART
train_dir = 'pdf/learning_set'
dictionary = make_dic(train_dir)
train_labels = np.zeros(20)
train_labels[17:20] = 1
train_matrix = extract_features(train_dir)
model1 = MultinomialNB()
model1.fit(train_matrix, train_labels)
# TESTING PART
test_dir = 'pdf/testing_set'
test_matrix = extract_features(test_dir)
test_labels = np.zeros(8)
test_labels[4:7] = 1
result1 = model1.predict(test_matrix)
print(confusion_matrix(test_labels, result1))
这是我的回溯:
Traceback (most recent call last):
File "ML.py", line 65, in <module>
model1.fit(train_matrix, train_labels)
File "/usr/local/lib/python3.6/site-packages/sklearn/naive_bayes.py",
line 579, in fit
X, y = check_X_y(X, y, 'csr')
File "/usr/local/lib/python3.6/site-
packages/sklearn/utils/validation.py", line 552, in check_X_y
check_consistent_length(X, y)
File "/usr/local/lib/python3.6/site-
packages/sklearn/utils/validation.py", line 173, in
check_consistent_length
" samples: %r" % [int(l) for l in lengths])
ValueError: Found input variables with inconsistent numbers of
samples: [23, 20]
我想知道如何解决这个问题? 我正在使用 Ubuntu 16.04 和 python 3.6。
最佳答案
ValueError:发现输入变量的数量不一致 样本:[23, 20]
这意味着您有 23 个训练向量(train_matrix 有 23 行) 但只有 20 个训练标签(train_labels 是一个包含 20 个值的数组)
更改train_labels = np.zeros(20)
到train_labels = np.zeros(23)
它应该可以工作。
关于python - python 中的机器学习 scikit learn 问题,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46786036/