mlflow attempt

Jenkinsfile_mlflow

@@ -31,6 +31,7 @@ pipeline {
                     sh 'python3 ./nn_train_mlflow.py'
                     archiveArtifacts artifacts: 'mlruns/**'
                     sh 'rm -r mlruns'
+                    sh 'rm -r my_model_mlflow'
                 }
             }
         }

nn_train_mlflow.py

@@ -83,6 +83,6 @@ with mlflow.start_run():
     mlflow.log_param("1st_activation_funct", first_activation_funct)
     mlflow.log_param("2nd_activation_funct", second_activation_funct)
     #mlflow.keras.log_model(model, 'my_model')
-    mlflow.keras.save_model(model, "my_model")
+    mlflow.keras.save_model(model, "my_model_mlflow")
     mlflow.log_metric("accuracy", accuracy_score(y_true, y_pred))