Guest User

Untitled

a guest
Oct 23rd, 2018
115
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.38 KB | None | 0 0
  1. with mlflow.start_run():
  2.  
  3. ... model ...
  4.  
  5. mlflow.log_param("source", wine_path)
  6. mlflow.log_param("alpha", alpha)
  7. mlflow.log_param("l1_ratio", l1_ratio)
  8.  
  9. mlflow.log_metric("rmse", rmse)
  10. mlflow.log_metric("r2", r2)
  11. mlflow.log_metric("mae", mae)
  12.  
  13. mlflow.set_tag('domain', 'wine')
  14. mlflow.set_tag('predict', 'quality')
  15. mlflow.sklearn.log_model(lr, "model")
Add Comment
Please, Sign In to add comment