sb-ai-lab / LightAutoML

Fast and customizable framework for automatic ML model creation (AutoML)
https://developers.sber.ru/portal/products/lightautoml
Apache License 2.0
1.18k stars 51 forks source link

Add built-in matching for predictions #68

Open dev-rinchin opened 1 year ago

dev-rinchin commented 1 year ago

🚀 Feature Request

Add matching function to map AutoML's predictions for multiclass task and call it before predict method return predictions. automl.reader.class_mapping is not convenient.