Fixed filtering of unk from transcript

DevKiHyun · Apr 26, 2017 · 4537c2a · 4537c2a
1 parent bfab1c3
commit 4537c2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/data_loader.py b/data/data_loader.py
@@ -94,7 +94,7 @@ def __getitem__(self, index):
  def parse_transcript(self, transcript_path):
  with open(transcript_path, 'r') as transcript_file:
  transcript = transcript_file.read().replace('\n', '')
- transcript = [self.labels_map.get(x, None) for x in list(transcript) if x is not None]
+ transcript = filter(None, [self.labels_map.get(x) for x in list(transcript)])
  return transcript
 
  def __len__(self):