DAminoMuta / vis /kcc_upd.py
auralray's picture
Upload folder using huggingface_hub
acbef3a verified
import pandas as pd
df = pd.DataFrame()
df['gt'] = pd.read_csv('../run-reg/gru-att-256-super-256-0.001-50/preds.csv')['gt']
df['GRU 256 ATT'] = pd.read_csv('../run-reg/gru-att-256-super-256-0.001-50/preds.csv')['fusion']
# df['GRU 128 ATT GF'] = pd.read_csv('../run-reg/gru-att-128-gf-super-256-0.001-50/preds.csv')['fusion']
df['GRU 256 ATT GF'] = pd.read_csv('../run-reg/gru-att-256-gf-super-256-0.001-50/preds.csv')['fusion']
# df['GRU 512 ATT GF'] = pd.read_csv('../run-reg/gru-att-512-gf-super-256-0.001-50/preds.csv')['fusion']
df['GRU 256 MLP'] = pd.read_csv('../run-reg/gru-mlp-256-super-256-0.001-50/preds.csv')['fusion']
df['GRU 256 MLP GF'] = pd.read_csv('../run-reg/gru-mlp-256-gf-super-256-0.001-50/preds.csv')['fusion']
df['MHA 256 ATT'] = pd.read_csv('../run-reg/mha-att-256-super-256-0.001-50/preds.csv')['fusion']
df['MHA 256 ATT GF'] = pd.read_csv('../run-reg/mha-att-256-gf-super-256-0.001-50/preds.csv')['fusion']
df['MHA 256 MLP'] = pd.read_csv('../run-reg/mha-mlp-256-super-256-0.001-50/preds.csv')['fusion']
df['MHA 256 MLP GF'] = pd.read_csv('../run-reg/mha-mlp-256-gf-super-256-0.001-50/preds.csv')['fusion']
df['LSTM 128 ATT'] = pd.read_csv('../run-reg/lstm-att-128-super-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 ATT'] = pd.read_csv('../run-reg/lstm-att-256-super-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 ATT GF'] = pd.read_csv('../run-reg/lstm-att-256-gf-super-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 MLP'] = pd.read_csv('../run-reg/lstm-mlp-256-super-256-0.001-50/preds.csv')['fusion']
df['LSTM 256 MLP GF'] = pd.read_csv('../run-reg/lstm-mlp-256-gf-super-256-0.001-50/preds.csv')['fusion']
df['LSTM 512 ATT'] = pd.read_csv('../run-reg/lstm-att-512-super-256-0.001-50/preds.csv')['fusion']
df['Mamba 128 ATT'] = pd.read_csv('../run-reg/mamba-att-128-super-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 ATT'] = pd.read_csv('../run-reg/mamba-att-256-super-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 ATT GF'] = pd.read_csv('../run-reg/mamba-att-256-gf-super-256-0.001-50/preds.csv')['fusion']
df['Mamba 512 ATT'] = pd.read_csv('../run-reg/mamba-att-512-super-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 MLP'] = pd.read_csv('../run-reg/mamba-mlp-256-super-256-0.001-50/preds.csv')['fusion']
df['Mamba 256 MLP GF'] = pd.read_csv('../run-reg/mamba-mlp-256-gf-super-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_regs.csv', index=False)