search

blobcity / autoai

Python based framework for Automatic AI for Regression and Classification over numerical data. Performs model search, hyper-parameter tuning, and high-quality Jupyter Notebook code generation.
Apache License 2.0
175 stars 43 forks source link

Add AdaBoostRegressor #60

Closed Thilakraj1998 closed 3 years ago

Thilakraj1998 commented 3 years ago

Add AdaBoostRegressor model into the library.

Primary File to Change: https://github.com/blobcity/autoai/blob/main/blobcity/config/regressor_config.py

Reference AdaBoostRegressor Implementation: https://github.com/blobcity/ai-seed/blob/main/Regression/Adaptive%20Boosting/AdaBoostRegressor.ipynb

Official API Refer for parameters: https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.AdaBoostRegressor.html

Dependencies if any, must be appropriately added. Test run of the train function on a regression problem must pass, and the function must attempt to train an AdaBoostRegressor as a potential best fit model.

26tanishabanik commented 3 years ago

@Thilakraj1998 For the adaboost regressor, do I need to define the base estimator as a list of object data type, like this "ensemble.DecisionTreeClassifier(max_depth=3)"?