NN_Classification

source

NN_Classification(
   data_obj: Classification_Data
)

Methods:

.get_model

source

.get_model(
   data_obj
)

---

Initialize the neural network

Args

• data_obj : Classification_Data object

Returns

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

.plot

source

.plot(
   data_obj
)

---

Creates the plots

Args

• data_obj : Classification_Data object

Returns

data_object with modified variables