Add experiment name

Author: ace-racer

train.py

@@ -44,6 +44,7 @@ def eval_metrics(actual, pred):
     alpha = float(sys.argv[1]) if len(sys.argv) > 1 else 0.5
     l1_ratio = float(sys.argv[2]) if len(sys.argv) > 2 else 0.5
 
+    mlflow.set_experiment("/wine-quality-experiment")
     with mlflow.start_run():
         lr = ElasticNet(alpha=alpha, l1_ratio=l1_ratio, random_state=42)
         lr.fit(train_x, train_y)