From 8aa04f9fac81d74389eb5ac033df0259c0469a34 Mon Sep 17 00:00:00 2001 From: Jan Nowak Date: Sun, 23 May 2021 15:37:37 +0200 Subject: [PATCH] Registred model. --- .gitignore | 3 ++- generate_MLmodel.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 26aacea..b7733bd 100644 --- a/.gitignore +++ b/.gitignore @@ -6,4 +6,5 @@ metrics.tsv plot.png my_runs mlruns -my_model \ No newline at end of file +my_model +1/ \ No newline at end of file diff --git a/generate_MLmodel.py b/generate_MLmodel.py index 77c6098..ed9ea6f 100644 --- 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)