s487194/ium_487194
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

mlflow fix

Browse Source
This commit is contained in:
Witold Woch 2023-05-14 18:13:25 +02:00
parent 73a958e269
commit aa20f3caa4
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
MLproject.yml
View File

@ -4,13 +4,12 @@ docker_env:
image: s487194/ium:1 image: s487194/ium:1
entry_points: entry_points:
main: train:
parameters: parameters:
epochs: {type: int, default: 100} epochs: {type: int, default: 100}
lr: {type: float, default: 0.01} lr: {type: float, default: 0.01}
command: "python train.py --epochs {epochs} --lr {lr}" command: "python train.py --epochs {epochs} --lr {lr}"
test: test:
parameters:
command: "python prediction.py" command: "python prediction.py"