[Python] 데이터 셋 간단하게 나누기

모델을 학습 하면서 데이터셋을 나누는 경우에 사용합니다.

pytorch의 dataloader같은 방법들도 있으나 데이터 프레임이나 리스트를 바로 나누어 볼때 사용하면 좋습니다.

from sklearn.model_selection import train_test_split

train, test = train_test_split(df, test_size=0.2)