This commit is contained in:
Witold Woch 2023-05-14 18:11:11 +02:00
parent 6aa9dfba3d
commit 73a958e269

16
MLproject.yml Normal file
View File

@ -0,0 +1,16 @@
name: s487194
docker_env:
image: s487194/ium:1
entry_points:
main:
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"