주 메뉴 열기
대문
임의의 문서로
로그인
설정
Pywiki 소개
면책 조항
Pywiki
검색
바뀜
← 이전 편집
Scikit-learn
(편집)
2023년 2월 5일 (일) 10:37 판
273 바이트 추가됨
,
2023년 2월 5일 (일) 10:37
→기초 사용법
29번째 줄:
29번째 줄:
|훈련데이터와 테스트데이터 분리
|훈련데이터와 테스트데이터 분리
|귀찮은 일을 모듈이 해준다.
|귀찮은 일을 모듈이 해준다.
+
{| class="wikitable"
+
|+자주 사용하는 옵션
+
!의도
+
!설명
+
|-
+
|stratify=df['해당열']
+
|범주별 비율을 비슷하게 맞춰 데이터를 분리한다.
+
(원 데이터에서 하나의 범주만 너무 많거나 적은 경우 사용하여 비율을 지킨다.)
+
|}
|<syntaxhighlight lang="python">
|<syntaxhighlight lang="python">
from sklearn.model_selection import train_test_split
from sklearn.model_selection import train_test_split
Sam
사무관
,
인터페이스 관리자
,
관리자
, 교사
편집
1,408
번