name: s444517_train
docker_env:
   image:  kambobdocker420/ium:mlflow
entry_points:
  main:
    parameters:
      epochs: {type: int, default: 200}
      first_activation_funct: {type: str, default: "relu"}
      second_activation_funct: {type: str, default: "softmax"}
    command: "python nn_train_mlflow.py {epochs} {first_activation_funct} {second_activation_funct}"