name: s478815
    
docker_env:
  image: korneliag/ium478815:1.0

entry_points:
  main:
    parameters:
      epochs: {type: float, default: 1000}
    command: "python train-mlfow.py {epochs}"