Machine Learning

[Machine Learning] LightBGM

joo_ 2022. 5. 17. 19:31

분류, 회귀 문제에 모두 사용할 수 있는 강력한 모델로,

부스팅 계열의 트리 모델

병렬 처리 성능과 자동 가지치기 알고리즘이 적용되어 XGBoost 대비 성능은 비슷하지만 속도가 훨씬 빠르다

 

!pip install lightgbm

from lightgbm import LGBMClassifier

lgbc = LGBMClassifier()
lgbc.fit(x_train, y_train)

[출처] https://leehah0908.tistory.com/18