battelle.nn.NeuralNetwork.train

NeuralNetwork.train(x, y, learning_rate=0.05)

Train the neural network with dataset of features x and of corresponding labels y. using the backpropagation algorithm.

Parameters
  • x (np.array) – A list of features

  • y (np.array) – The corresponding list of labels.