diff --git a/downloads-generation/models_class1/GENERATE.sh b/downloads-generation/models_class1/GENERATE.sh index ae98f51ebe9a09381c3dd497d49dfcdd2643dfdb..a38faa4887715f7abf7a865074062126f224f074 100755 --- a/downloads-generation/models_class1/GENERATE.sh +++ b/downloads-generation/models_class1/GENERATE.sh @@ -31,7 +31,7 @@ time mhcflurry-class1-train-allele-specific-models \ --data "$(mhcflurry-downloads path data_curated)/curated_training_data.csv.bz2" \ --hyperparameters hyperparameters.yaml \ --out-models-dir models \ - --min-measurements-per-allele 200 + --min-measurements-per-allele 100 cp $SCRIPT_ABSOLUTE_PATH . bzip2 LOG.txt diff --git a/downloads-generation/models_class1/hyperparameters.yaml b/downloads-generation/models_class1/hyperparameters.yaml index 0e114b320231a585d8ef78df925caff74dbf43c6..30b78fa22c370b316c989b32c59064ff8aed938c 100644 --- a/downloads-generation/models_class1/hyperparameters.yaml +++ b/downloads-generation/models_class1/hyperparameters.yaml @@ -2,7 +2,7 @@ ########################################## # ENSEMBLE SIZE ########################################## -"n_models": 12, +"n_models": 8, ########################################## # OPTIMIZATION @@ -32,11 +32,6 @@ # NEURAL NETWORK ARCHITECTURE ########################################## "locally_connected_layers": [ - { - "filters": 8, - "activation": "tanh", - "kernel_size": 3 - }, { "filters": 8, "activation": "tanh",