name: Fifa Players
docker_env:
   image:  docker.io/adnovac/ium_s434760:2.0 
entry_points:
  train:
    parameters:
      batch_size: {type: int, default: 15}
      epochs: {type: int, default: 16}
    command: "python train.py {batch_size} {epochs}"
  evaluate:
    command: "python evaluate.py"