[머신러닝 커닝 페이퍼] 분류를 위한 결정트리모델의 모든 것 1편
from sklearn.tree import DecisionTreeClassifier from sklearn.model_selection import train_test_split from sklearn.metrics import accuracy_score # 학습데이터와 테스트데이터 나눠준다 X_train, X_test, y_train, y_test = train_test_split(data, label, test_size = 0.2, random_state=0) # 결정트리 모델 학습 dt_clf = DecisionTreeClassifier(random_state=0) #dt_clf = DecisionTreeClassifier(random_state=0, min_samples_split = 3, ma..
프로그래밍/Data&ML
2022. 12. 1. 18:54