【发布时间】:2019-12-03 03:52:38
【问题描述】:
获取异常Exception: Data must be 1-dimensional
在 Python 3.7 中使用 NumPy
相同的代码适用于其他人,但不适用于我的情况。下面是我的代码,请帮忙
import numpy as np
from sklearn import linear_model
from sklearn.model_selection import train_test_split
import seaborn as sns
from sklearn import metrics
import matplotlib.pyplot as plt
%matplotlib inline
df = pd.read_csv('./Data/new-data.csv', index_col=False)
x_train, x_test, y_train, y_test = train_test_split(df['Hours'], df['Marks'], test_size=0.2, random_state=42)
sns.jointplot(x=df['Hours'], y=df['Marks'], data=df, kind='reg')
x_train = np.reshape(x_train, (-1,1))
x_test = np.reshape(x_test, (-1,1))
y_train = np.reshape(y_train, (-1,1))
y_test = np.reshape(y_test, (-1,1))
#
print('Train - Predictors shape', x_train.shape)
print('Test - Predictors shape', x_test.shape)
print('Train - Target shape', y_train.shape)
print('Test - Target shape', y_test.shape)
预期的输出应该是
Train - 预测器形状 (80, 1)
测试 - 预测器形状 (20, 1)
训练 - 目标形状 (80, 1)
测试 - 目标形状 (20, 1)
作为输出得到异常Exception: Data must be 1-dimensional
【问题讨论】:
-
您的第一个
jpg暗示了问题,系列上的reshape将被弃用。
标签: python-3.x numpy regression linear-regression