Udacity: 作业3:ValueError:错误的输入形状 (1000, 10)

我正在完成作业3:正则化。在查看了Github上的内容后,我尝试自己解决这个作业,但遇到了运行时错误。请注意,我选择的数据集大小比链接中提到的要小。

这是当前的情况:

print('Training set', train_dataset.shape, train_labels.shape)print('Validation set', valid_dataset.shape, valid_labels.shape)print('Test set', test_dataset.shape, test_labels.shape)#Training set (20000, 784) (20000, 10)#Validation set (1000, 784) (1000, 10)#Test set (1000, 784) (1000, 10)

问题出在这里:

from sklearn.linear_model import LogisticRegressionoriginal_train_labels = train_labelslogit_clf = LogisticRegression(penalty='l2')logit_clf.fit(train_dataset[:1000,:], original_train_labels[:1000])

运行后会得到以下错误:

---------------------------------------------------------------------------ValueError                                Traceback (most recent call last)<ipython-input-12-4888dc0bbc75> in <module>()      4       5 logit_clf = LogisticRegression(penalty='l2')----> 6 logit_clf.fit(train_dataset[:1000,:], original_train_labels[:1000])      7 predicted = logit_clf.predict(test_dataset)      8 print('accuracy', accuracy((np.arange(num_labels) == predicted[:,None]).astype(np.float32), test_labels), '%')/usr/local/lib/python2.7/dist-packages/sklearn/linear_model/logistic.pyc in fit(self, X, y, sample_weight)   1140    1141         X, y = check_X_y(X, y, accept_sparse='csr', dtype=np.float64, -> 1142                          order="C")   1143         check_classification_targets(y)   1144         self.classes_ = np.unique(y)/usr/local/lib/python2.7/dist-packages/sklearn/utils/validation.pyc in check_X_y(X, y, accept_sparse, dtype, order, copy, force_all_finite, ensure_2d, allow_nd, multi_output, ensure_min_samples, ensure_min_features, y_numeric, warn_on_dtype, estimator)    513                         dtype=None)    514     else:--> 515         y = column_or_1d(y, warn=True)    516         _assert_all_finite(y)    517     if y_numeric and y.dtype.kind == 'O':/usr/local/lib/python2.7/dist-packages/sklearn/utils/validation.pyc in column_or_1d(y, warn)    549         return np.ravel(y)    550 --> 551     raise ValueError("bad input shape {0}".format(shape))    552     553 ValueError: bad input shape (1000, 10)

有什么解决方法吗?


回答:

你对train_labels使用了独热编码。这意味着它的形状是[1000, 10],有1000个样本,每个样本有10个’列’,其中1表示我们正在讨论的类别。这对于神经网络是必需的,但sklearn的Logistic Regression 要求它的形状是[1000, 1],也就是说,它应该是一个有1000行的向量,每行应该有一个整数来指示目标类别。使用argmax函数将独热编码转换为整数,你的问题就解决了。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注