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 SGDRegressor #62

Closed Thilakraj1998 closed 3 years ago

Thilakraj1998 commented 3 years ago

Add SGDRegressor model into the library.

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

Reference SGDRegressor Implementation: https://github.com/blobcity/ai-seed/blob/main/Regression/Linear%20Models/SGDRegressor.ipynb

Offical API Reference for Parameter: https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.SGDRegressor.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 SGDRegressor as a potential best fit model.

melan96 commented 3 years ago

I will go through it. I'll the PR as soon as possible!

melan96 commented 3 years ago

Commit #82 with PR added for solve this issue.