나무구조로 도표화하여 분류/예측을 수행하는 머신러인 알고리즘 (지도학습)
개념/코드가 잘 정리되어있는 글이 있어 첨부합니다.
analysis-flood.tistory.com/100
scikit-learn.org/stable/modules/tree.html
DecisionTree Classifiier VS DecisionTree Regression
-
Decision Tree Classifier: 분류 문제를 해결하는데 사용. 타켓변수가 범주형일 때 사용. 의사결정나무의 terminal node에는 타겟변수인 범주형 값들이 출력된다. 예) 대출 승인을 받을지 여부 예측
-
Decision Tree Regressor: 회귀 문제를 해결하는데 사용. 타켓변수가 연속형일 때 사용. 의사결정나무의 terminal node에는 타겟변수인 연속형 값들이 출력된다. 예)마취과다로 인해 얼마나 많은 사람이 죽을지를 예측
becominghuman.ai/machine-learning-series-day-6-decision-tree-regressor-82a2e2f873a
DecisionTreeRegressor 변수 설명
class sklearn.tree.DecisionTreeRegressor(*, criterion='mse', splitter='best', max_depth=None, min_samples_split=2, min_samples_leaf=1, min_weight_fraction_leaf=0.0, max_features=None, random_state=None, max_leaf_nodes=None, min_impurity_decrease=0.0, min_impurity_split=None, ccp_alpha=0.0)
scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeRegressor.html
'Stay Hungry Stay Foolish > ML & DL' 카테고리의 다른 글
[파이토치 첫걸음] 4. 인공신경망 (0) | 2021.09.28 |
---|---|
Faster R-CNN 구현 (0) | 2021.02.19 |
[밑바닥부터 시작하는 딥러닝] 4장 신경망 학습 (2) | 2021.02.01 |
[Coursera Deep Learning] Neural Networks and Deep Learning 강의 추천 (0) | 2021.01.26 |
[밑바닥부터 시작하는 딥러닝] 정리 링크 공유 (2) | 2021.01.24 |
댓글