Skip to content

Instantly share code, notes, and snippets.

@kelvintaywl

kelvintaywl/test.py

Created February 16, 2019 07:14
Show Gist options
  • Save kelvintaywl/36e19b8ce9957d2f9c6a5f683dfa9f8c to your computer and use it in GitHub Desktop.
Save kelvintaywl/36e19b8ce9957d2f9c6a5f683dfa9f8c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
test.py
mlpRegressor = MLPRegressor(hidden_layer_sizes=100, activation="relu", solver="lbfgs")
mlpRegressor.fit(train_data, y_train)
rmse_score, r2_score = test(mlpRegressor, test_data, test_ans)
print(f'score: RMSE={rmse_score:.4f}, R2={r2_score:.4f}')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment