ium_487176/MLProject
2023-06-28 13:23:09 +02:00

11 lines
273 B
Plaintext

name: s487176
docker_env:
image: mactyc/ium
entry_points:
train:
parameters:
num_epochs {type: int, default: 10}
lr : {type: int, default : 0.01}
alpha: {tpe : int, default: 0.01}
command: "python zad1.py --num_epochs {num_epochs} --lr {lr}"