Dionyssos commited on
Commit
27ac763
1 Parent(s): ccb2aa8
Files changed (1) hide show
  1. visualize_per_sentence.py +8 -1
visualize_per_sentence.py CHANGED
@@ -66,10 +66,17 @@ preds = {}
66
 
67
  for file_interface in FULL_PKL:
68
  y = pd.read_pickle(file_interface)
69
- preds[file_interface] = y
 
 
 
 
70
 
71
 
72
 
 
 
 
73
  for lang in ['english',
74
  'foreign']:
75
 
 
66
 
67
  for file_interface in FULL_PKL:
68
  y = pd.read_pickle(file_interface)
69
+ # y = y.rolling(20).mean()[19:] --> avoid when printing character error rate
70
+ preds[file_interface] = y #.sort_values('styletts2-valence')
71
+ print(f'\n\n {file_interface}\n_____________________________\n',
72
+ f"{y['cer-prompt'].mean()=}",
73
+ f"{y['cer-styletts2'].mean()=}\n\n")
74
 
75
 
76
 
77
+ # =================================== cER ---------------------------
78
+
79
+
80
  for lang in ['english',
81
  'foreign']:
82