티스토리 뷰
모델을 학습 하면서 데이터셋을 나누는 경우에 사용합니다.
pytorch의 dataloader같은 방법들도 있으나 데이터 프레임이나 리스트를 바로 나누어 볼때 사용하면 좋습니다.
from sklearn.model_selection import train_test_split
train, test = train_test_split(df, test_size=0.2)
'기계학습' 카테고리의 다른 글
| [Pytorch]“AttributeError: LSTM object has no attribute flat_weights_names” (0) | 2020.03.31 |
|---|---|
| [Pytorch] F.cross entropy vs nn.Cross Entropy loss (0) | 2020.02.27 |
| 경량 딥러닝 간단한 흐름 정리 (0) | 2019.09.06 |
| [세팅] window10 tensorflow gpu 설치하기 (1) | 2019.08.23 |
| [keras] fit vs fit_generator (1) | 2019.02.13 |
댓글