Sklearn train_test_split()

1 개요

Sklearn train_test_split()

import numpy as np
from sklearn.model_selection import train_test_split
X, y = np.arange(10).reshape((5, 2)), range(5)
print( "X=\n", X )
print( "y=", list(y) )
print( "--------------------" )
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
print( "X_train=\n", X_train )
print( "y_train=", y_train )
print( "--------------------" )
print( "X_test=\n", X_test)
print( "y_test=", y_test)

2 같이 보기

• 데이터 분할
• 파이썬 데이터 분할