jmquintana79
12/12/2017 - 5:09 AM

split train test using cv

Split dataset in recursive Train/Test datasets using cross validation.

split_TrainTest_cv.py
content_copyfile_download
from sklearn.model_selection import KFold

X = np.array([[1, 2], [3, 4], [1, 2], [3, 4]]) # create an array
y = np.array([1, 2, 3, 4]) # Create another array

kf = KFold(n_splits=2, random_state=None, shuffle=False)
nf = kf.get_n_splits(X) # returns the number of splitting iterations in the cross-validator
for train_index, test_index in kf.split(X):
    print('TRAIN:', train_index, 'TEST:', test_index)
    X_train, X_test = X[train_index], X[test_index]
    y_train, y_test = y[train_index], y[test_index]