Skip to content
Snippets Groups Projects
Commit 7cd4b795 authored by Julia K's avatar Julia K
Browse files

If output models directory doesn't exist, create it

parent c4f42dd3
No related branches found
No related tags found
No related merge requests found
......@@ -2,6 +2,7 @@
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")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment