Autogluon: Tabular: Add XGBoost Model

Created on 30 Jul 2020  路  11Comments  路  Source: awslabs/autogluon

Add XGBoost model to increase model diversity in AutoGluon-Tabular.

enhancement feature request help wanted tabular

All 11 comments

HI @Innixma
Can I work on this?

HI @Innixma
Can I work on this?

Sure! Please reference the existing LightGBM and CatBoost implementations when making your model class, and reference example_custom_model_tabular.py for an example of adding a custom model to AutoGluon.

@Dipanks Do you have an update on this? If not I may consider taking up the task.

Hi, @Innixma Would I join this work? 馃槃
I am interested in model diversity in AutoGluon-Tabular.
hope to start from this XGBoost which use similar API style with sklearn

@sackoh Hi David, feel free to take up this task and reference the existing LightGBM and Catboost implementations to aid in your code. Do note that I will plan to take up this task in a couple weeks if a contribution has not been submitted, as it is high priority, so please consider the timeline in which you plan to work on this, as submitting a PR within the next 1-2 weeks would be highly preferred.

@Innixma Ok, I consider this in my priority.

@Innixma Hi, Nick!
I have spent a couple of days to review the code structure of tabular task.
I started to work on this task in my forked repo.
Hope to finish within coming next 1-2 weeks. 馃

Awesome, thanks for the update!

@Innixma Hi, Nick. I have cretaed #691 to add XGBoost model.

Great, I'll take a look!

XGBoost model has now been merged in. Thanks @sackoh for the contribution!

Was this page helpful?
0 / 5 - 0 ratings