change for Domino2

Jenkinsfile

@@ -33,7 +33,7 @@ pipeline {
                 {
                        sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
                        sh 'rm -r my_model'
-                       sh 'python3 vgsales-mlflow.py ${BATCH_SIZE} ${EPOCHS}'
+                       sh 'python3 vgsales-mlflow.py'
                 }
             }
 

vgsales-mlflow.py

@@ -66,4 +66,3 @@ with mlflow.start_run():
     mlflow.log_metric("rmse", rmse) 
     #mlflow.keras.log_model(model, 'vgsales_model.h5')
     mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train)
-