Ожидаемый двумерный массив, вместо ошибки есть скалярный массив

Question

Я работаю над python 3.7. Я получаю сообщение об ошибке при выполнении кода ниже. Как я могу это решить?

import pandas as pd
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

data = pd.read_csv("hw_25000.csv")
regression = LinearRegression()
boy= data.Height.values.reshape(-1,1)
kilo= data.Weight.values.reshape(-1,1)

regression.fit(boy,kilo)
regression.predict(70)

ошибка:

ValueError: Ожидаемый 2D-массив, вместо него получен скалярный массив: array = 1. Измените ваши данные, используя array.reshape (-1, 1), если ваши данные имеют один элемент, или array.reshape (1, -1), если он содержит один образец.

hw_25000.csv :

Index,Height,Weight
1, 65.78331, 112.9925
2, 71.51521, 136.4873
3, 69.39874, 153.0269
4, 68.2166, 142.3354

Answer 1

Вы не можете предсказать int, это должен быть массив

reg.predict(np.array(70).reshape(-1, 1))
array([[141.94045785]])