인프런 커뮤니티 질문&답변

tiny123450님의 프로필 이미지
tiny123450

작성한 질문수

[개정판] 파이썬 머신러닝 완벽 가이드

LightGBM을 이용한 위스콘신 유방암 예측

LightGBM 4.0.0을 이용하고 있는데 수업에서 사용한 3.3.2버전과 다른점이 있나요?

작성

·

1.1K

0

'LightGBM을 이용한 위스콘신 유방암 예측'을 보고

깃허브에서 다운한 예제를 런 시켜보고 있는데

세번째에서 오류가 뜹니다. fit()함수에 early_stopping_rounds는 쓸수 없다고 나옵니다. 그래서 파라미터명이 업데이트 되었나 싶어서 인터넷에 찾아보니까 LightGBM 페이지에서는 early_stopping_rounds, early_stopping_round, 둘다 가능하다고 나옵니다. 뭐가 문제인지 모르겠습니다.

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까,

음, lightgbm에서 4.0 신 버전에서 early_stopping을 callback 호출 방식으로 변경한 것 같군요. 이게 아직 사이킷런 Lightgbm 래퍼에 잘 안맞는것 같습니다.

lightgbm 사이킷런 래퍼에서 fit()이 early stopping으로 적용하는 방법이 아직 명확하지 않은 것 같습니다(제가 방법을 찾지 못했습니다).

현재 실습 코드에서 수행이 되려면 lightgbm이 downgrade 되어야 할 것 같습니다.

pip install lightgbm == 3.3.2 로 다운하셔야 수행이 될 것 같습니다.

감사합니다.

tiny123450님의 프로필 이미지
tiny123450

작성한 질문수

질문하기