diff --git a/train.py b/train.py index 2cd66be..99f5b7b 100755 --- a/train.py +++ b/train.py @@ -21,7 +21,7 @@ logging.basicConfig(level=logging.WARN) logger = logging.getLogger(__name__) mlflow.set_tracking_uri("http://172.17.0.1:5000") -mlflow.set_experiment("s123456-new") +mlflow.set_experiment("s123456-2023") def eval_metrics(actual, pred): rmse = np.sqrt(mean_squared_error(actual, pred))