최대 1 분 소요

※ Scikit-Learn

  • 머신러닝에 특화된 라이브러리
# e.g.
# sklearn을 사용하여 분류 모델을 학습, 평가하는 예시

# 모델 불러오기
> from sklearn.ensemble import RandomForestClassifier
> from sklearn.metrics import accuracy_score

# 데이터 분할
> from sklearn.model_selection import train_test_split
> X_train, X_test, y_train, y_test = train_test_split(X, y)

# 모델 객체 선언
> model = RandomForestClassifier()

# training data로 학습 진행
> model.fit(X_train, y_train)

# test data로 inference 진행
> pred = model.predict(X_test)

# Evaluation metric으로 평가 진행
> print("Accuracy : %.4f" accuracy_score(y_test, pred)) # Accuracy : ~~~

댓글남기기