Diff of /tasks_base/generate.py [000000] .. [5d1c0a]

Switch to side-by-side view

--- a
+++ b/tasks_base/generate.py
@@ -0,0 +1,26 @@
+'''Training script generator for subject-independent classification.
+'''
+with open('script.sh.template', 'r') as f:
+    sh_template = f.read()
+with open('train.pbs.template', 'r') as f:
+    pbs_template = f.read()
+
+
+def list_to_str(l):
+    return ' '.join(map(str, l))
+
+
+all_folds = list(range(54))
+command_template = "python train_base.py $datapath $outpath -fold {0} -gpu {1} > $logpath/stdout.f{0}.out &"
+for i, j in enumerate(range(0, len(all_folds), 8)):
+    folds = all_folds[j:j+8]
+    commands = []
+    for ind, fold in enumerate(folds):
+        commands.append(command_template.format(fold, ind))
+    commands.append('wait')
+    content = sh_template + '\n'.join(commands)
+    with open('script.' + str(i) + '.sh', 'w') as f:
+        f.write(content)
+    content = pbs_template.format(i)
+    with open('train.' + str(i) + '.pbs', 'w') as f:
+        f.write(content)