train_test_split: ValueError

Question

В настоящее время я пытаюсь вычислить среднеквадратичную ошибку конкретной модели полиномиальной регрессии. Однако, когда я запускаю train_test_split, я получаю сообщение об ошибке:

ValueError: too many values to unpack (expected 2)

Вот код, который я написал:

def mse(X, y, degree, model):
    X_train, y_train = train_test_split(X, y, test_size=0.6, random_state=10)
    train_errors = []
    for m in range(1, len(X_train)):
        model.fit(X_train[:m], y_train[:m])
        y_train_predict = model.predict(X_train[:m])
        mse = train_errors.append(mean_squared_error(y_train[:m], y_train_predict))
    return mse

Я не уверен, почему я получаю эту ошибку или что я могу изменить, чтобы заставить код работать! Любой совет будет оценен!

Answer 1

train_test_split также возвращает тестовый образец:

X_train, X_test, y_train, y_test = train_test_split(
        X, y, test_size=0.6, random_state=10)