If output models directory doesn't exist, create it
Train Class1 single allele models.
"""
+import os
import sys
import argparse
import json
@@ -93,6 +94,11 @@ def run(argv=sys.argv[1:]):
predictor = Class1AffinityPredictor()
+ if args.out_models_dir and not os.path.exists(args.out_models_dir):
+ print("Attempting to create directory: %s" % args.out_models_dir)
+ os.mkdir(args.out_models_dir)
+ print("Done.")
+
for (h, hyperparameters) in enumerate(hyperparameters_lst):
n_models = hyperparameters.pop("n_models")
