search

stanfordmlgroup / ngboost

Natural Gradient Boosting for Probabilistic Prediction
Apache License 2.0
1.62k stars 214 forks source link

ngboost uses sklearn's DecisionTreeClassifier to report errors #304

Closed qq492947833 closed 1 year ago

qq492947833 commented 1 year ago

When I use ngboost's classification function, if base calls sklearn's DecisionTreeClassifier, it gets an error: ValueError: Unknown label type: 'continuous'. My trainy already uses astype('int') when using the fit feature, but I still get this error.

当我使用ngboost的分类功能时，如果base调用sklearn的DecisionTreeClassifier，会报错：ValueError: Unknown label type: 'continuous'。在使用fit功能时，我的trainy已经使用了astype('int')，但是仍然会出现这个错误。

qq492947833 commented 1 year ago

Sorry, #225 solved my problem,i will close this question！