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数组，而是标量数组：   数组= 5000。   如果数据具有单个功能，则使用array.reshape（-1，1）来重塑数据；如果包含单个样本，则使用array.reshape（1，-1）来重塑数据。

Answer 1

更改行：

model.predict(5000)

至：

model.predict([[5000]])