def predict(self, data): nn = NeuralNetwork() l1 = Layer(56, 54) l2 = Layer(54, 25) nn.add(l1) nn.add(ActivationLayer(relu, relu_derivative)) nn.add(l2) nn.add(ActivationLayer(sigmoid, sigmoid_derivative)) l1.weights = np.load('weights1.npy') l2.weights = np.load('weights2.npy') l1.bias = np.load('bias1.npy') l2.bias = np.load('bias2.npy') out = nn.predict(data) pred = np.argmax(out) return pred