name: iumKC

conda_env: environment.yaml

entry_points:
  main:
    command: "python train.py --train=true --save_model=true"
  test:
    parameters:
      model_filepath: "model"
    command: "python test.py --test=true --load_model=$model_filepath"