Commit 44955a48 authored by Nicole Bussola's avatar Nicole Bussola
Browse files

manage single modality

parent b0211eed
......@@ -55,7 +55,12 @@ df_results = pd.DataFrame(columns=['layers', 'mcc_train','mcc_train_min','mcc_tr
'sens_train','sens_train_min','sens_train_max', 'spec_train', 'spec_train_min', 'spec_train_max',
'mcc_test', 'mcc_test_min', 'mcc_test_max', 'best_feat'])
for k in range(2, N_LAYERS + 1):
if MODE=='single':
range_combinations = range(1,2)
else:
range_combinations = range(2, N_LAYERS + 1)
for k in range_combinations:
for comb in combinations(LAYERS, k):
......@@ -160,5 +165,5 @@ for k in range(2, N_LAYERS + 1):
df_results = df_results.append(row, ignore_index=True)
df_results.to_csv(f'{OUTFOLDER}/{DATASET}/{TARGET}/{MODEL}/metrics_allSplits_{MODE}.txt', sep='\t', index=False)
df_results.to_csv(f'{OUTFOLDER}/{DATASET}/{TARGET}/{MODEL}/metrics_allSplits_{MODE}.txt', sep='\t', index=False)
# %%
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment