replace bash script to docker-compose to build server

2023-03-07 16:33:45 +08:00
parent ec19042d0d
commit 1f86146b12
7 changed files with 28 additions and 16 deletions
--- a/train.py
+++ b/train.py
@ -64,7 +64,7 @@ if __name__ == '__main__':
    optimizer = SGD(model.parameters(), lr=learning_rate)

    # set the tracking URI to the model registry
-    mlflow.set_tracking_uri('http://127.0.0.1:5000')
+    mlflow.set_tracking_uri('http://127.0.0.1:5001')
    mlflow.set_experiment('train_fortune_predict_model')

    # start a new MLflow run