diff --git a/downloads-generation/models_class1_minimal/README.md b/downloads-generation/models_class1_minimal/README.md new file mode 100644 index 0000000000000000000000000000000000000000..ddf90f672cd4ac9e69dd941c831e30b5bef445e5 --- /dev/null +++ b/downloads-generation/models_class1_minimal/README.md @@ -0,0 +1,13 @@ +# Class I allele-specific models (minimal: ensemble size of 1) + +This download contains "minimal" MHC Class I MHCflurry predictors consisting +of a single model per allele. These predictors are expected to have slightly +lower accuracy than the standard ensembles (models_class1) but are small and +fast. Useful for testing. + +To download these models and set them as the default predictor, run: + +``` +$ mhcflurry-downloads fetch models_class1_minimal +$ export MHCFLURRY_DEFAULT_CLASS1_MODELS=$(mhcflurry-downloads path models_class1_minimal)/models +``` \ No newline at end of file diff --git a/mhcflurry/downloads.yml b/mhcflurry/downloads.yml index 6ca5cfb6a23c67c4b71b0fc1768eb67ff06792ef..09930b7aa6505e3e578729039572c666ce355737 100644 --- a/mhcflurry/downloads.yml +++ b/mhcflurry/downloads.yml @@ -32,6 +32,10 @@ releases: url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_unselected.20180221.tar.bz2 default: false + - name: models_class1_minimal + url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_minimal.20180226.tar.bz2 + default: false + - name: models_class1_consensus url: https://github.com/openvax/mhcflurry/releases/download/pre-1.2/models_class1_consensus.tar.bz2 default: false diff --git a/test/test_train_and_related_commands.py b/test/test_train_and_related_commands.py index 513c95dd93545f8f4bf4c141fda56ffc29e68a63..a43b6d2feb9b119fc956a10ba3ca957612b395b8 100644 --- a/test/test_train_and_related_commands.py +++ b/test/test_train_and_related_commands.py @@ -18,7 +18,7 @@ HYPERPARAMETERS = [ { "n_models": 2, "max_epochs": 500, - "patience": 10, + "patience": 5, "minibatch_size": 128, "early_stopping": True, "validation_split": 0.2,