Skip to content

Commit

Permalink
Update - push last changes
Browse files Browse the repository at this point in the history
  • Loading branch information
ymarghi committed Feb 16, 2024
1 parent d922440 commit 075b866
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions mmidas/eval_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,8 +94,6 @@ def summarize_inference(cpl_mixVAE, files, data, saving_folder=''):
consensus.append(armA_vs_armB_norm)

n_pruned.append(len(nprune_indx))
# category_vs_class = category_vs_class[:, :, nprune_indx]
# cT_vs_cT.append(category_vs_class)
plt.close()

data_dic = {}
Expand All @@ -106,7 +104,6 @@ def summarize_inference(cpl_mixVAE, files, data, saving_folder=''):
data_dic['con_mean'] = consensus_mean
data_dic['num_pruned'] = n_pruned
data_dic['pred_label'] = label_pred
# data_dic['cT_vs_cT'] = cT_vs_cT
data_dic['consensus'] = consensus
data_dic['armA_vs_armB'] = AvsB
data_dic['prune_indx'] = prune_indx
Expand Down

0 comments on commit 075b866

Please sign in to comment.