
name: NerBlackBox project

entry_points:

  run_experiment:
    parameters:
      experiment_name: {type: string}
      from_config: {type: int}
      run_name: {type: string}
      device: {type: string}
      fp16: {type: int}
    command: |
        python modules/scripts/script_run_experiment.py \
        --experiment_name {experiment_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}
