Skip to content

Instantly share code, notes, and snippets.

@susanli2016

susanli2016/neural_network

Created September 29, 2018 16:32
Show Gist options
  • Save susanli2016/f880e4ba87cf05167a60f9a58b5ffa43 to your computer and use it in GitHub Desktop.
Save susanli2016/f880e4ba87cf05167a60f9a58b5ffa43 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
neural_network
def get_neural_network(hidden_layer_size=50):
mlp = MLPClassifier(hidden_layer_sizes=hidden_layer_size)
return [mlp], ['MultiLayer Perceptron']
classifier_list, classifier_name_list = get_neural_network()
for classifier,classifier_name in zip(classifier_list,classifier_name_list):
classifier.fit(X_train,y_train)
print_evaluation_metrics(classifier,classifier_name,X_test,y_test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment