<딥러닝, 머신러닝> classify할 대상을 다양하게 train/test set에 넣고 싶다면?

Tensorflow의 경우,

sklearn.model_selection.train_test_split

 

함수가 존재한다. 그렇기 때문에 수월하게 train, test를 나눌 수 있다.

 

그 중 stratify함수를 사용하면, 분류할 대상이 한쪽으로 쏠리지 않고 다양하게 분류할 수 있다.

 

즉 stratify = classify할 target(label)으로 넣으면 된다.

 

https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html

sklearn.model_selection.train_test_split

sklearn 공홈에서 참고했다.