diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py index 81716fb..18442ff 100644 --- a/vgsales-mlflow.py +++ b/vgsales-mlflow.py @@ -16,6 +16,8 @@ from tensorflow.keras.models import Sequential import mlflow +mlflow.set_tracking_uri("http://172.17.0.1:5000") +mlflow.set_experiment("s434749") def my_main(epochs, batch_size):