diff --git a/downloads-generation/models_class1_pan_variants/GENERATE.sh b/downloads-generation/models_class1_pan_variants/GENERATE.sh index 7195099c1be297ad79c130a52126fbc36b0a9379..e36df1d71194d7c374a6498a9493579d26a7080b 100755 --- a/downloads-generation/models_class1_pan_variants/GENERATE.sh +++ b/downloads-generation/models_class1_pan_variants/GENERATE.sh @@ -69,7 +69,7 @@ then python generate_hyperparameters.py hyperparameters.production.yaml compact_peptide > hyperparameters.compact_peptide.yaml fi -VARIANTS=( no_additional_ms_ms_only_0nm ms_only_0nm no_additional_ms_0nm 0nm no_additional_ms no_pretrain compact_peptide 34mer_sequence single_hidden_no_pretrain affinity_only ) +VARIANTS=( no_additional_ms_ms_only_0nm ms_only_0nm no_additional_ms_0nm 0nm 500nm no_additional_ms no_pretrain compact_peptide 34mer_sequence single_hidden_no_pretrain affinity_only ) for kind in "${VARIANTS[@]}" do @@ -127,6 +127,16 @@ do HYPERPARAMETERS=hyperparameters.production.yaml fi + if [ "$kind" == "500nm" ] + then + TRAINING_DATA="train_data.$kind.csv" + python reassign_mass_spec_training_data.py \ + "$(mhcflurry-downloads path data_curated)/curated_training_data.csv.bz2" \ + --set-measurement-value 500 \ + --out-csv "$TRAINING_DATA" + HYPERPARAMETERS=hyperparameters.production.yaml + fi + if [ "$kind" == "ms_only_0nm" ] then TRAINING_DATA="train_data.$kind.csv"