Registred model.

2021-05-23 15:37:37 +02:00 · 2021-05-23 15:37:37 +02:00 · 8aa04f9fac
commit 8aa04f9fac
parent 63cb9a7d84
2 changed files with 3 additions and 2 deletions
--- a/.gitignore
+++ b/.gitignore
@ -7,3 +7,4 @@ plot.png
 my_runs
 mlruns
 my_model
+1/
--- a/generate_MLmodel.py
+++ b/generate_MLmodel.py
@ -29,7 +29,7 @@ if __name__ == "__main__":

    lr = args.lr
    n_epochs = args.epochs
-    mlflow.set_tracking_uri("http://127.0.0.1:5000")
+    mlflow.set_tracking_uri("http://172.17.0.1:5000")
    mlflow.set_experiment("s426206")
    with mlflow.start_run():
        mlflow.log_param("lr", lr)