ardaatahan commited on
Commit
5e78dfc
·
1 Parent(s): 82994f7

SW-202 add more descriptive comments and fix spelling

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -116,11 +116,11 @@ model_to_multilingual_wer = dict(
116
  # Add English WER and Multilingual WER to performance_df
117
  benchmark_df["english_wer"] = benchmark_df["model"].map(model_to_english_wer)
118
  benchmark_df["multilingual_wer"] = benchmark_df["model"].map(model_to_multilingual_wer)
119
- benchmark_df.fillna({"multilingual_wer": "N/A"}, inplace=True)
120
 
121
- # Set multilingual_wer to "English Only Model" for English-only models
122
  english_only_mask = benchmark_df["model"].str.contains(r"\.en$|distil-whisper", case=False, na=False)
123
- benchmark_df.loc[english_only_mask, "multilingual_wer"] = "English Only Model"
124
 
125
  benchmark_df["multilingual_wer"] = benchmark_df["multilingual_wer"].astype(str)
126
 
 
116
  # Add English WER and Multilingual WER to performance_df
117
  benchmark_df["english_wer"] = benchmark_df["model"].map(model_to_english_wer)
118
  benchmark_df["multilingual_wer"] = benchmark_df["model"].map(model_to_multilingual_wer)
119
+ benchmark_df.fillna({"multilingual_wer": "N/A"}, inplace=True) # Mark all untested models as N/A
120
 
121
+ # Mark English-only models
122
  english_only_mask = benchmark_df["model"].str.contains(r"\.en$|distil-whisper", case=False, na=False)
123
+ benchmark_df.loc[english_only_mask, "multilingual_wer"] = "English-only model"
124
 
125
  benchmark_df["multilingual_wer"] = benchmark_df["multilingual_wer"].astype(str)
126