Table 3 Model and library/class.
Model | Library/Class |
|---|---|
Decision tree | “sklearn.tree.DecisionTreeRegressor” |
LR | “sklearn.linear_model.LinearRegression” |
RR | “sklearn.linear_model.Ridge” |
Lasso regression | “sklearn.linear_model.Lasso” |
SVR | “sklearn.svm.SVR” |
RF | “sklearn.ensemble.RandomForestRegressor” |
GB | “sklearn.ensemble.GradientBoostingRegressor” |
AdaBoost | “sklearn.ensemble.AdaBoostRegressor” |
XGBoost | “xgboost.XGBRegressor” |
