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 LightGBMRegressor #59

Closed Thilakraj1998 closed 3 years ago

Thilakraj1998 commented 3 years ago

Add LightGBMRegressor model into the library.

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

Reference LightGBMRegressor Implementation: https://github.com/blobcity/ai-seed/blob/main/Regression/LightGBM/LGBMRegressor.ipynb

Also Refer offical documentation to select appropriate parameters: https://lightgbm.readthedocs.io/en/latest/pythonapi/lightgbm.LGBMRegressor.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 a LightGBMRegressor as a potential best fit model.

SaharshLaud commented 3 years ago

Can you assign this issue to me @Thilakraj1998 sir? I will work on it and provide the required code.