-
질문 & 답변
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
LightGBM 4.0.0을 이용하고 있는데 수업에서 사용한 3.3.2버전과 다른점이 있나요?
23.08.09 20:43 작성 조회수 818
0
'LightGBM을 이용한 위스콘신 유방암 예측'을 보고
깃허브에서 다운한 예제를 런 시켜보고 있는데
세번째에서 오류가 뜹니다. fit()함수에 early_stopping_rounds는 쓸수 없다고 나옵니다. 그래서 파라미터명이 업데이트 되었나 싶어서 인터넷에 찾아보니까 LightGBM 페이지에서는 early_stopping_rounds, early_stopping_round, 둘다 가능하다고 나옵니다. 뭐가 문제인지 모르겠습니다.
답변을 작성해보세요.
0
권 철민
지식공유자2023.08.10
안녕하십니까,
음, lightgbm에서 4.0 신 버전에서 early_stopping을 callback 호출 방식으로 변경한 것 같군요. 이게 아직 사이킷런 Lightgbm 래퍼에 잘 안맞는것 같습니다.
lightgbm 사이킷런 래퍼에서 fit()이 early stopping으로 적용하는 방법이 아직 명확하지 않은 것 같습니다(제가 방법을 찾지 못했습니다).
현재 실습 코드에서 수행이 되려면 lightgbm이 downgrade 되어야 할 것 같습니다.
pip install lightgbm == 3.3.2 로 다운하셔야 수행이 될 것 같습니다.
감사합니다.
답변 1