NN_Regression

source

NN_Regression(
   data_obj: Regression_Data
)

---

Neural Network 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(
   data_obj
)

---

Initialize the neural network

Args

• data_obj : Regression_Data object

Returns

tf.keras.Sequential model (via self.model)

.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