[python] colab pycaret 설치

코랩에서 여러 모델의 성능을 비교해볼 수 있는 pycaret을 사용하기 위해 설치를 하려고 

기존에 다른 패키지를 설치하는 것처럼 설치했더니,, 

!pip install pycaret

오류가 발생한다.. 

 

sklearn 버전에 맞게 재설치

우선 설치된 sklearn의 버전이 0.23.2로 맞추어져야 한다고 오류가 떠서 

버전에 맞게 sklearn을 삭제 및 재설치 해준다

!pip uninstall sklearn -y
!pip install --upgrade sklearn
!pip install scikit-learn==0.23.2 --user

위 코드를 실행시킨 다음 런타임 재실행 후 다시 sklearn의 버전을 확인하여야 0.23.2 버전으로 설치된 것을 확인할 수 있다

import sklearn
sklearn.__version__

 

colab에서 pycaret 설치

from pycaret.utils import enable_colab
enable_colab()

!pip install pycaret

from pycaret.classification import *

분류모델을 비교해야 해서 위와 같이 classification으로 전부를 import 하였다

 

pycaret 사용

exp = setup(data = data, target = 'target') #전체 데이터와 y값 지정

아래의 모델들을 추가해서 모델 비교 진행

catboost = create_model('catboost')
mlp = create_model('mlp')
xgboost = create_model('xgboost')

models()