Lv1 | 모델링 | scikit-learn (DecisionTreeClassifier)
Data Analyst

빅데이터 관련 자료/Dacon

Lv1 | 모델링 | scikit-learn (DecisionTreeClassifier)

carpe08 2021. 7. 28. 16:02
320x100
320x100
import sklearn
from sklearn.tree import DecisionTreeClassifier

EDA를 통해 데이터를 살펴보고, 전처리를 하였다면, 본격적으로 머신러닝 모델을 훈련시키고, 훈련된 모델을 통해 예측을 한다.
선행 연구된 놀랍고 다양한 모델들이 있는데, 이러한 머신러닝 모델들을 TensorFlow, PyTorch 등 Python 오픈 소스 머신 러닝 라이브러를 통해 손쉽게 구현 가능하다
먼저 sckit-learn 라이브러리를 사용해 모델링을 시작한다.


scikit-learn 홈페이지: https://scikit-learn.org/stable/

320x100
320x100