<딥러닝, 머신러닝> classify할 대상을 다양하게 train/test set에 넣고 싶다면?
2021. 10. 29. 23:56ㆍ전공공부/전자전기전공
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: Release Highlights for scikit-learn 0.23 Release Highlights for scikit-learn 0.23, Release Highlights for scikit-learn 0.24 Release Highligh...
scikit-learn.org
sklearn 공홈에서 참고했다.
'전공공부 > 전자전기전공' 카테고리의 다른 글
| 오디오 신호처리 for deep learning - FFT, STFT, spectrogram, scalogram (1) | 2022.04.14 |
|---|---|
| tf.keras.layers.Dense의 Args 분석 - TNT 스터디 중 몰랐던 것 (0) | 2021.09.17 |
| ConcatOp: Dimensions of inputs should match shape ... error가 나올 때 (0) | 2021.09.17 |
| TNT 스터디 2주차 몰랐던 것(1) (0) | 2021.09.17 |
| VScode, Colab에서 명령어가 입력 되지 않을 때 한가지 방법 (0) | 2021.09.16 |