如何畫出GradientBoostingClassifier的learning curve

2019/07/23 下午 09:48

機器學習共學討論版

黃品叡

觀看數：23

回答數：1

收藏數：0

大家好!

我想利用learning curve畫出GradientBoostingClassifier在訓練時的學習曲線，其程式碼如下:

然而其運行結果出現以下錯誤:

我自己的解讀是，這個1與0的分類問題在進行權重分配後，使得0這個分類的權重變為零，因而造成只有一個分類。我查過GradientBoostingClassifier與learning_curve的document,並未發現有調整sample_weight的參數。

請問我該如何解決這個問題呢?