train script

kundajelab · Jan 29, 2024 · 5a08e5b · 5a08e5b
1 parent d368438
commit 5a08e5b
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/chrombpnet/train_bias.py b/chrombpnet/train_bias.py
@@ -0,0 +1,19 @@
+import parsers as parsers
+import os
+from data import DefaultDataFile, get_default_data_path
+from data import print_meme_motif_file
+import pipelines as pipelines
+import copy
+import pandas as pd
+import logging
+logging.getLogger('matplotlib.font_manager').disabled = True
+
+
+args = parsers.read_parser()
+os.makedirs(os.path.join(args.output_dir,"logs"), exist_ok=True)
+os.makedirs(os.path.join(args.output_dir,"auxiliary"), exist_ok=True)
+os.makedirs(os.path.join(args.output_dir,"models"), exist_ok=True)
+os.makedirs(os.path.join(args.output_dir,"evaluation"), exist_ok=True)
+
+pipelines.train_bias_pipeline(args)
+
diff --git a/chrombpnet/train_chrombpnet.py b/chrombpnet/train_chrombpnet.py
@@ -0,0 +1,18 @@
+import parsers as parsers
+import os
+from data import DefaultDataFile, get_default_data_path
+from data import print_meme_motif_file
+import pipelines as pipelines
+import copy
+import pandas as pd
+import logging
+logging.getLogger('matplotlib.font_manager').disabled = True
+
+args = parsers.read_parser()
+
+os.makedirs(os.path.join(args.output_dir,"logs"), exist_ok=True)
+os.makedirs(os.path.join(args.output_dir,"auxiliary"), exist_ok=True)
+os.makedirs(os.path.join(args.output_dir,"models"), exist_ok=True)
+os.makedirs(os.path.join(args.output_dir,"evaluation"), exist_ok=True)
+
+pipelines.chrombpnet_train_pipeline(args)