mlflow fix

2023-05-14 18:13:25 +02:00 · 2023-05-14 18:13:25 +02:00 · aa20f3caa4
commit aa20f3caa4
parent 73a958e269
1 changed files with 2 additions and 3 deletions
--- a/MLproject.yml
+++ b/MLproject.yml
@ -4,13 +4,12 @@ docker_env:
  image: s487194/ium:1

 entry_points:
-  main:
+  train:
    parameters:
      epochs: {type: int, default: 100}
      lr: {type: float, default: 0.01}
    command: "python train.py --epochs {epochs} --lr {lr}"
+    
  test:
-    parameters:
-      
    command: "python prediction.py"