<딥러닝, 머신러닝> 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 Examples using sklearn.model_selection.train_test_split..
2021.10.29