Updated testing suite (completed testing, so should delete when able)

AutoResearch · Feb 12, 2024 · a67c656 · a67c656
1 parent 41f16b5
commit a67c656
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/testing_suite/TESTING-AE-GAN.docx b/testing_suite/TESTING-AE-GAN.docx
diff --git a/testing_suite/gan_visualize_erps_main.py b/testing_suite/gan_visualize_erps_main.py
@@ -4,8 +4,8 @@
 import torch
 
 #### Define parameters ####
-gan_filename = 'aegan_ep8000_p100_e2_enc25-1.pt'
-data_filename = 'ganTrialElectrodeERP_p100_e2_len100.csv'
+gan_filename = 'gan_ddp_ep4000_p100_e8.pt'
+data_filename = 'ganTrialElectrodeERP_p100_e8_len100.csv'
 
 #### Load generated data ####
 syn_filename = gan_filename.replace('.pt','')
@@ -60,4 +60,22 @@
     ax[1,int(electrode-1)].get_xaxis().set_visible(False)
     ax[1,int(electrode-1)].get_yaxis().set_visible(False)
     ax[1,int(electrode-1)].spines[['right', 'top']].set_visible(False)
+
+    '''
+    #Empirical
+    ax[int(electrode-1)].plot(c0_emp[c0_emp['Electrode']==electrode].mean()[emp_data_index:], label='C0')
+    ax[int(electrode-1)].plot(c1_emp[c1_emp['Electrode']==electrode].mean()[emp_data_index:], label='C1')
+    ax[int(electrode-1)].set_title(f'Empirical (E: {int(electrode)})')
+    ax[int(electrode-1)].get_xaxis().set_visible(False)
+    ax[int(electrode-1)].get_yaxis().set_visible(False)
+    ax[int(electrode-1)].spines[['right', 'top']].set_visible(False)
+
+    #Synthetic
+    ax[int(electrode)].plot(c0_syn[c0_syn['Electrode']==electrode].mean()[syn_data_index:], label='C0')
+    ax[int(electrode)].plot(c1_syn[c1_syn['Electrode']==electrode].mean()[syn_data_index:], label='C1')
+    ax[int(electrode)].set_title(f'Synthetic (E: {int(electrode)})')
+    ax[int(electrode)].get_xaxis().set_visible(False)
+    ax[int(electrode)].get_yaxis().set_visible(False)
+    ax[int(electrode)].spines[['right', 'top']].set_visible(False)
+    '''
 plt.show()