name: s478815
docker_env:
image: korneliag/ium478815:1.0
entry_points:
main:
parameters:
epochs: {type: float, default: 1000}
command: "python train-mlfow.py {epochs}"