
name: NerBlackBox project

entry_points:

  run_training:
    parameters:
      training_name: {type: string}
      from_config: {type: int}
      run_name: {type: string}
      device: {type: string}
      fp16: {type: int}
    command: |
        python modules/scripts/script_run_training.py \
        --training_name {training_name} \
        --from_config {from_config} \
        --run_name {run_name} \
        --device {device} \
        --fp16 {fp16}

  set_up_dataset:
    parameters:
      ner_dataset: {type: string}
      ner_dataset_subset: {type: string}
      modify: {type: int}
      val_fraction: {type: float}
      verbose: {type: int}
    command: |
        python modules/scripts/script_set_up_dataset.py \
        --ner_dataset {ner_dataset} \
        --ner_dataset_subset {ner_dataset_subset} \
        --modify {modify} \
        --val_fraction {val_fraction} \
        --verbose {verbose}

  analyze_data:
    parameters:
      ner_dataset: {type: string}
      verbose: {type: int}
    command: |
        python modules/scripts/script_analyze_data.py \
        --ner_dataset {ner_dataset} \
        --verbose {verbose}
