Затяжка метод в Python

голоса
0

Как сделать 6: 4 затяжки в Python? Я попытался следующий код:

X_train, X_test, y_train, y_test =  train_test_split(X,y, training_size=0.6, test_size=0.4)

Но не уверен, является ли это правильно или нет.

Задан 09/10/2019 в 12:55
источник пользователем
На других языках...                            


1 ответов

голоса
2

Обратитесь к документации train_test_splitот scikit учиться.

Для того, чтобы установить как размер обучения и тестовый набор нужно передать train_size(не training_sizeкак в вашем коде) и test_size.

Для того, чтобы использовать 60% ваши данные для обучения и 40% для тестирования вы можете использовать это:

import numpy as np
from sklearn.model_selection import train_test_split

X = np.random.rand(100, 2)
y = range(100)

X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.6, test_size=0.4)

Вы можете подтвердить, что для 100 точек данных, используемых в этом примере вы получаете поезд набор размера 60 и тестовый набор размер 40:

print(len(X_train), len(X_test))
print(len(y_train), len(y_test))
> 60 40
> 60 40
Ответил 09/10/2019 в 13:53
источник пользователем

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more