diff --git a/README.md b/README.md index 585d6b3d..15255756 100644 --- a/README.md +++ b/README.md @@ -1 +1,22 @@ # Example MLflow project + + +## Run Locally + +Clone the project + +```bash + git clone https://github.com/mlflow/mlflow-example.git +``` + +Run the project and track the metrics with mlflow ui + +```bash + mlflow ui +``` + +make sure you are inside the mlflow-example directory +```bash + mlflow run . +``` + diff --git a/train.py b/train.py index 26765faf..f9d9cb23 100644 --- a/train.py +++ b/train.py @@ -57,7 +57,7 @@ def eval_metrics(actual, pred): print(" MAE: %s" % mae) print(" R2: %s" % r2) - mlflow.log_param("alpha", alpha) + mlflow.log_param("alpha", int(alpha)) mlflow.log_param("l1_ratio", l1_ratio) mlflow.log_metric("rmse", rmse) mlflow.log_metric("r2", r2)