错误:形状 (1,4) 和 (14,14) 不对齐

我是一个机器学习的新手,对这个错误感到有些困惑:

形状 (1,4) 和 (14,14) 不对齐:4 (维度1) != 14 (维度0)

这是完整的错误信息:

文件 “/Users/jim/anaconda3/lib/python3.6/site-packages/sklearn/utils/extmath.py”, 第140行,safe_sparse_dot函数中返回 np.dot(a, b)

数值错误:形状 (1,4) 和 (14,14) 不对齐:4 (维度1) != 14 (维度0)

我的测试集有4行数据,训练集有14行数据,正如(1,4)和(14,14)所示。至少我认为是这个意思。

我试图对训练集进行简单的线性回归拟合,如下面的代码所示:

# 对训练集进行简单线性回归拟合from sklearn.linear_model import LinearRegressionregressor = LinearRegression()X_train = X_train.reshape(1,-1)y_train = y_train.reshape(1,-1)regressor.fit(X_train, y_train)

然后预测测试集结果:

# 预测测试集结果X_test = X_test.reshape(1,-1)y_pred = regressor.predict(X_test)

我的代码在最后一行出现了上述错误:

y_pred = regressor.predict(X_test)

任何正确的指导都会非常有帮助。

这是我的完整代码样本:

# 简单线性回归# 导入库import numpy as npimport matplotlib.pyplot as pltimport pandas as pd# 导入数据集dataset = pd.read_csv('NBA.csv')X = dataset.iloc[:, 1].valuesy = dataset.iloc[:, :-1].values# 将数据集拆分为训练集和测试集from sklearn.cross_validation import train_test_splitX_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)# 特征缩放# 无# 对训练集进行简单线性回归拟合from sklearn.linear_model import LinearRegressionregressor = LinearRegression()X_train = X_train.reshape(1,-1)y_train = y_train.reshape(1,-1)regressor.fit(X_train, y_train)# 预测测试集结果X_test = X_test.reshape(1,-1)y_pred = regressor.predict(X_test)

** 编辑 **我检查了X和y的形状。以下是我的输出:

dataset = pd.read_csv('NBA.csv')X = dataset.iloc[:, 1].valuesy = dataset.iloc[:, :-1].valuesprint(X.shape)print(y.shape)-->(18,)-->(18, 1)

回答:

请将所有reshape(1,-1)替换为reshape(-1, 1)。前者将数组转换为(1人 x n特征),后者转换为(n人 x 1特征)。在本例中,特征身高

如果您将导入部分修改为如下所示,则无需重新调整数组的形状,因为它们的形状已经符合(n人 x 1特征)的形式。

# 导入数据集dataset = pd.read_csv('NBA.csv')X = dataset.iloc[:, 1].valuesy = dataset.iloc[:, 0].valuesX = X.reshape(-1, 1)y = y.reshape(-1, 1)

在sklearn的早期版本中,您可以将向量作为输入。但最近已有变化,现在您需要明确指出向量是(1样本 x n特征)还是(n样本 x 1特征),可以通过使用reshape或其他方法来实现。

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中创建了一个多类分类项目。该项目可以对…

发表回复

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