nubank / fklearn

fklearn: Functional Machine Learning
Apache License 2.0
1.51k stars 165 forks source link

Support new multi-class objectives in lgbm_classification_learner #232

Closed IgorKhomyanin closed 1 year ago

IgorKhomyanin commented 1 year ago

Status

READY

Todo list

Background context

The lgbm_classification_learner supported only a multiclass objective for a multi-class classification problem.

When a softmax (alias for multiclass) or a multiclassova (and all its aliases) were used as an objective, lgbm_classification_learner failed to produce a prediction function, resulting in an error.

Description of the changes proposed in the pull request

Where should the reviewer start?

Changes were only made inside lgbm_classification_learner.

caique-lima commented 1 year ago

Closing and reopening to trigger CI