name: tutorial

docker_env:
  image: karopa/ium:27

entry_points:
  main:
    parameters:
      epochs: {type: float, default: 30}
    command: "python3 neural_network.py {epochs}"
  test:
    command: "python3 evaluate_network.py"