【发布时间】:2016-05-31 01:00:06
【问题描述】:
我正在尝试使用 sklearn 并使用 sklearn 库在 Python 中执行线性回归。
这是我用来训练和拟合模型的代码,当我运行预测函数调用时出现错误。
train, test = train_test_split(h1, test_size = 0.5, random_state=0)
my_features = ['bedrooms', 'bathrooms', 'sqft_living', 'sqft_lot', 'floors', 'zipcode']
trainInp = train[my_features]
target = ['price']
trainOut = train[target]
regr = LinearRegression()
# Train the model using the training sets
regr.fit(trainInp, trainOut)
print('Coefficients: \n', regr.coef_)
testPred = regr.predict(test)
拟合模型后,当我尝试使用测试数据进行预测时,会抛出以下错误
Traceback (most recent call last):
File "C:/Users/gouta/PycharmProjects/MLCourse1/Python.py", line 52, in <module>
testPred = regr.predict(test)
File "C:\Users\gouta\Anaconda2\lib\site-packages\sklearn\linear_model\base.py", line 200, in predict
return self._decision_function(X)
File "C:\Users\gouta\Anaconda2\lib\site-packages\sklearn\linear_model\base.py", line 183, in _decision_function
X = check_array(X, accept_sparse=['csr', 'csc', 'coo'])
File "C:\Users\gouta\Anaconda2\lib\site-packages\sklearn\utils\validation.py", line 393, in check_array
array = array.astype(np.float64)
ValueError: invalid literal for float(): 20140604T000000
线性回归模型的系数是
('Coefficients: \n', array([[ -5.04902429e+04, 5.23550164e+04, 2.90631319e+02,
-1.19010351e-01, -1.25257545e+04, 6.52414059e+02]]))
以下是测试数据集的前五行
是因为系数值大导致的错误吗?如何解决这个问题?
【问题讨论】:
-
为什么值中有
T这个字母?另外,考虑展示你的一些代码...... -
请向我们展示实际引发错误的代码。
-
你能显示
test的前几行吗? -
"这个错误是因为系数值大吗?" 请向我们展示测试数据的前几行!
-
对不起。我犯的错误是,我为训练输入和目标选择了许多列,但是测试数据集中的所有列都包含在测试数据集中,因此测试数据集中还有其他变量导致了问题。
标签: python scikit-learn linear-regression sklearn-pandas