text_plain_output_1.png
from sklearn.neural_network import MLPClassifier

(X_train.shape, y_train.shape, X_test.shape, y_test.shape)

from sklearn.neural_network import MLPClassifier
mlp_classifier = MLPClassifier(learning_rate_init=0.0001, random_state=0, max_iter=300, hidden_layer_sizes=(256, 64)).fit(X_train, y_train)

print('Evaluate:')
print(f'train score: {mlp_classifier.score(X_train, y_train)}')
print(f'test socre: {mlp_classifier.score(X_test, y_test)}')