Логистическая регрессия

Question

Мои данные делятся на обучающие и тестируемые из набора данных df. Могу ли я разделить поезд и проверить как X_train, Y_train из поезда и X_test, Y_test из теста?

Пример кода: -

train, test = train_test_split(df, train_size=0.8, stratify=df.y_yes.values, random_state=5)

#Can I do the below splitting?
X_train, Y_train = train_test_split(train, random_state=5)
X_test, Y_test = train_test_split(test, random_state=5)

train.head и test.head

Answer 1

Я не мог найти решение этой проблемы, поэтому использовал регулярное выражение разбиения: -

 X = df_clean[['previous', 'emp.var.rate', 'euribor3m', 'nr.employed', 
'pdays_missing', 'poutcome_success' , 'poutcome_nonexistent','pdays_bet_5_15']]
 y = df_clean['y_yes']
 X_train, X_test, Y_train, Y_test = train_test_split(X, y, test_size = 0.2, 
 random_state=5)
 print(X_train.shape)
 print(X_test.shape)
 print(Y_train.shape)
 print(Y_test.shape)