Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- with mlflow.start_run():
- ... model ...
- mlflow.log_param("source", wine_path)
- mlflow.log_param("alpha", alpha)
- mlflow.log_param("l1_ratio", l1_ratio)
- mlflow.log_metric("rmse", rmse)
- mlflow.log_metric("r2", r2)
- mlflow.log_metric("mae", mae)
- mlflow.set_tag('domain', 'wine')
- mlflow.set_tag('predict', 'quality')
- mlflow.sklearn.log_model(lr, "model")
Add Comment
Please, Sign In to add comment