ann_best_number_hidden.py

import sklearn.neural_network
import numpy
import matplotlib.pyplot

input_data = numpy.array([[1, 4],
                          [4, 1],
                          [3, 2],
                          [2, 1]])
output_data = numpy.array([0, 0, 1, 1])

my_neural_network = sklearn.neural_network.MLPClassifier(hidden_layer_sizes=(2,), max_iter=1000)

my_neural_network.fit(X=input_data, y=output_data)

labels = my_neural_network.predict(X=input_data)
print("Predicted Labels:", labels)