ValueError в файле sklearn's linear_model.predict

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), если он содержит один образец.

Answer 1

изменить строку:

model.predict(5000)

на:

model.predict([[5000]])