DAminoMuta / vis /auroc_upd.py
auralray's picture
Upload folder using huggingface_hub
acbef3a verified
import pandas as pd
df = pd.read_csv('./preds_results.csv')
df['gt'] = pd.read_csv('../run-cls/gru-att-256-ce-256-0.001-50/preds.csv')['gt']
df['GRU 256 ATT'] = pd.read_csv('../run-cls/gru-att-256-ce-256-0.001-50/preds.csv')['fusion']
df['GRU 256 ATT GF'] = pd.read_csv('../run-cls/gru-att-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['GRU 256 MLP'] = pd.read_csv('../run-cls/gru-mlp-256-ce-256-0.001-50/preds.csv')['fusion']
df['GRU 256 MLP GF'] = pd.read_csv('../run-cls/gru-mlp-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['MHA 256 ATT'] = pd.read_csv('../run-cls/mha-att-256-ce-256-0.001-50/preds.csv')['fusion']
df['MHA 256 ATT GF'] = pd.read_csv('../run-cls/mha-att-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['MHA 256 MLP'] = pd.read_csv('../run-cls/mha-mlp-256-ce-256-0.001-50/preds.csv')['fusion']
df['MHA 256 MLP GF'] = pd.read_csv('../run-cls/mha-mlp-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['LSTM 128 ATT'] = pd.read_csv('../run-cls/lstm-att-128-ce-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 ATT'] = pd.read_csv('../run-cls/lstm-att-256-ce-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 ATT GF'] = pd.read_csv('../run-cls/lstm-att-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 MLP'] = pd.read_csv('../run-cls/lstm-mlp-256-ce-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 MLP GF'] = pd.read_csv('../run-cls/lstm-mlp-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['LSTM 512 ATT'] = pd.read_csv('../run-cls/lstm-att-512-ce-256-0.001-50/preds.csv')['fusion']
df['Mamba 128 ATT'] = pd.read_csv('../run-cls/mamba-att-128-ce-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 ATT'] = pd.read_csv('../run-cls/mamba-att-256-ce-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 ATT GF'] = pd.read_csv('../run-cls/mamba-att-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['Mamba 512 ATT'] = pd.read_csv('../run-cls/mamba-att-512-ce-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 MLP'] = pd.read_csv('../run-cls/mamba-mlp-256-ce-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 MLP GF'] = pd.read_csv('../run-cls/mamba-mlp-256-gf-ce-256-0.001-50/preds.csv')['fusion']
df['DS R1'] = pd.read_csv('../dsr1/results_1_1.csv')['MIC_change']
df['DS R1 Mamba Fusion'] = pd.read_csv('../dsr1/results_2_1.csv')['MIC_change']
df.to_csv('./preds_results.csv', index=False)