RF_Regression

source

RF_Regression(
   data_obj: Regression_Data
)

---

RandomForest Regression.

Args

• data_obj : Regression_Data object

Returns

data_obj with filled result variables

Methods:

.run_regressor

source

.run_regressor(
   data_obj
)

---

Load or create a model, train model (if applicable), make predictions for trained model or uploaded model if it matches the data. Evaluate and plot results

Args

• data_obj : Regression_Data object

Returns

data_obj with modified values

.train_model

source

.train_model()

---

Initialize the random forest

Returns

sklearn RandomForestRegressor

.evaluate

source

.evaluate(
   data_obj
)

---

Create the evaluation, R2 Score, MAE and MSE

Args

• data_obj : Regression_Data object

Returns

data_obj with modified values

.plot

source

.plot(
   data_obj
)

---

Creates the output plots

Args

• data_obj : Regression_Data object

Returns

data_obj with modified values