지방이의 Data Science Lab

[python] 데이터 셋 쪼개기 유저기준으로 본문

Data Analysis/Python

[python] 데이터 셋 쪼개기 유저기준으로

[지현] 2019. 8. 4. 04:27
1
2
3
4
5
6
7
from sklearn.model_selection import train_test_split
from sklearn.model_selection import GroupShuffleSplit
 
train_inds, test_inds=next(GroupShuffleSplit(test_size=.2,n_splits=2,random_state=7).split(data,groups=data['acc_id']))
train=data.iloc[train_inds]
test=data.iloc[test_inds]
 

 

저작자표시 비영리 동일조건

'Data Analysis > Python' 카테고리의 다른 글

[python] nested list를 하나의 리스트로  (0) 2019.08.12
[python] 이미지삽입  (0) 2019.08.10
[python] 데이터 타입 바꾸기  (0) 2019.07.29
[python] 시간컬럼을 카테고리로 변경  (0) 2019.07.29
[python] array속 비교  (0) 2019.07.28
'Data Analysis/Python' Related Articles more
Comments