search

sagnik1511 / Tabular-AutoML

Python Auto-ML Package for Tabular Datasets
MIT License
23 stars 13 forks source link

[Issue 6] Added new models for regression training #39

Closed PalaVenkiReddy closed 1 year ago

PalaVenkiReddy commented 1 year ago

Added Decision Tree Regressor : For small training datasets training it with Random forest regressor takes more computation time without much better results when compared to Decision Tree Regressor. Added LGBMRegressor : It is capable of performing equally good with large datasets with a significant reduction in training time as it supports parallel learning.

sagnik1511 commented 1 year ago

P.S. Whenever a new PR is created, make sure to tag the issue number to ease the review process for the reviewer.

PalaVenkiReddy commented 1 year ago

Sorry Bhaiya, for mentioning issue number properly, I will update it.