diff --git a/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py b/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py index fd5172a5708789ee52004f86cce5f7c068069a13..e8a67737eb989c0a6df1142abd7a9d0a40703e71 100644 --- a/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py +++ b/mhcflurry/class1_affinity_prediction/train_allele_specific_models_command.py @@ -14,7 +14,7 @@ import mhcnames from .class1_binding_predictor import Class1BindingPredictor - +from ..common import configure_logging def normalize_allele_name(s): try: @@ -65,6 +65,8 @@ parser.add_argument( def run(): args = parser.parse_args(sys.argv[1:]) + configure_logging(verbose=args.verbosity > 1) + hyperparameters_lst = json.load(open(args.hyperparameters)) assert isinstance(hyperparameters_lst, list) print("Loaded hyperparameters list: %s" % str(hyperparameters_lst))