Ошибка значения при попытке разделить обучающий и тестовый набор

#python #statistics

#python #Статистика

Вопрос:

Я попробовал следующий код на python:

 X=df[["age","bmi","children"]].values.reshape(-1,1)
Y=df["charges"].values.reshape(-1,1)
from sklearn.preprocessing import MinMaxScaler
X_scaler = MinMaxScaler()
Y_scaler = MinMaxScaler()
X = X_scaler.fit_transform(X)
Y = Y_scaler.fit_transform(Y)
X_train, X_test, y_train, y_test = train_test_split(
    X, Y, test_size=0.2, random_state=1)
 

Но это дает мне следующую ошибку:

 ValueError: Found input variables with inconsistent numbers of samples: [2136, 712]
 

Если я сделаю это без изменения формы, это даст мне:

 ValueError: Expected 2D array, got 1D array instead
 

Так что я не знаю, что делать
Если кто-то может мне помочь, я буду признателен

Комментарии:

1. добавьте dataset в свой вопрос

Ответ №1:

Ну, я нашел решение, я изменил Y:

 Y = Y_scaler.fit_transform(Y.values.reshape(-1,1))
 

И это сработало