Assume you have a set of feature (features
) and labels (labels
).
Using this code base, we will create a Neural Network instance.
NN = Neural_Network()
We can then perform forward propagation using the .forward
method.
NN.forward(features)
Note: For the existing code base, the Neural Network class has specified values for input layers, hidden layers, and output layers.
Again, assume you have a set of feature (features
) and labels (labels
).
Using this code base, we will create a trainer instance.
T = trainer(NN)
We can then train the model on our data using the .train
method.
T.train(features, labels)
Note: You need to pass the Neural Network instance to create a trainer instance.