From 0ae03a70d4c365f38ccac66650e49fea888747be Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Wed, 14 Feb 2018 10:52:37 -0500
Subject: [PATCH] fix

---
 downloads-generation/models_class1_unselected/GENERATE.sh | 1 -
 mhcflurry/class1_neural_network.py                        | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/downloads-generation/models_class1_unselected/GENERATE.sh b/downloads-generation/models_class1_unselected/GENERATE.sh
index f9f3545c..a3cdb5c0 100755
--- a/downloads-generation/models_class1_unselected/GENERATE.sh
+++ b/downloads-generation/models_class1_unselected/GENERATE.sh
@@ -38,7 +38,6 @@ PROCESSORS=$(getconf _NPROCESSORS_ONLN)
 echo "Detected processors: $PROCESSORS"
 
 time mhcflurry-class1-train-allele-specific-models \
-    --allele HLA-A*02:01 HLA-A*01:01 HLA-A*03:01 HLA-A*11:01 HLA-A*24:02 HLA-B*07:02 HLA-B*15:01 \
     --data "$(mhcflurry-downloads path data_curated)/curated_training_data.no_mass_spec.csv.bz2" \
     --hyperparameters hyperparameters.yaml \
     --out-models-dir models \
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 4bc8331b..9b284122 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -103,7 +103,7 @@ class Class1NeuralNetwork(object):
     """
 
     miscelaneous_hyperparameter_defaults = HyperparameterDefaults(
-        train_data={'subset': 'all', 'num_poitns': None},
+        train_data={'subset': 'all'},
     )
     """
     Miscelaneous hyperaparameters. These parameters are not used by this class
-- 
GitLab