python - sklearn 的 linear_model.predict 上的 ValueError

Question

import numpy as np 
import pandas as pd
from sklearn import datasets, linear_model
df = pd.read_csv("homeprices.csv")
df
model = linear_model.LinearRegression()
model.fit(df[['area']], df.price)
model.predict(5000)

ValueError：预期的 2D 数组，得到了标量数组：array=5000。如果您的数据具有单个特征，则使用 array.reshape(-1, 1) 重塑您的数据，如果它包含单个样本，则使用 array.reshape(1, -1) 。

标签： pythonscikit-learndata-sciencevalueerror

Answer 1

换行：

model.predict(5000)

至：

model.predict([[5000]])