## Get the samples
X_samples = X[0:5]
print("Sample input:\n", X_samples)

# Do a forward pass of the network
output = model.forward(X_samples)
print("\nNetwork output:\n", output)

# Predict the label of each point
y_predicted = model.predict(X_samples)
print("\nPredicted labels:\n", y_predicted)