From 2e488acce1299bede9178ddeb245d67252dd3d1c Mon Sep 17 00:00:00 2001 From: akalankag Date: Thu, 6 Oct 2022 11:33:35 -0600 Subject: [PATCH] dict comp is faster than invoking the dict constructor --- Chapter03/MachineTranslation.py | 18 +++++++++--------- Chapter03/MachineTranslation_word2vec.py | 18 +++++++++--------- 2 files changed, 18 insertions(+), 18 deletions(-) diff --git a/Chapter03/MachineTranslation.py b/Chapter03/MachineTranslation.py index b29faf2..811b5ba 100644 --- a/Chapter03/MachineTranslation.py +++ b/Chapter03/MachineTranslation.py @@ -87,15 +87,15 @@ def vocab_generation(self,path,num_samples,verbose=True): print('Number of unique output tokens:', self.num_decoder_words) print('Max sequence length for inputs:', self.max_encoder_seq_length) print('Max sequence length for outputs:', self.max_decoder_seq_length) - - self.input_word_index = dict( - [(word, i) for i, word in enumerate(input_words)]) - self.target_word_index = dict( - [(word, i) for i, word in enumerate(target_words)]) - self.reverse_input_word_dict = dict( - (i, word) for word, i in self.input_word_index.items()) - self.reverse_target_word_dict = dict( - (i, word) for word, i in self.target_word_index.items()) + + self.input_word_index = { + word: i for i, word in enumerate(input_words)} + self.target_word_index = { + word: i for i, word in enumerate(target_words)} + self.reverse_input_word_dict = { + i: word for word, i in self.input_word_index.items()} + self.reverse_target_word_dict = { + i: word for word, i in self.target_word_index.items()} def process_input(self,input_texts,target_texts=None,verbose=True): diff --git a/Chapter03/MachineTranslation_word2vec.py b/Chapter03/MachineTranslation_word2vec.py index e3f4ea5..fa19c70 100644 --- a/Chapter03/MachineTranslation_word2vec.py +++ b/Chapter03/MachineTranslation_word2vec.py @@ -87,15 +87,15 @@ def vocab_generation(self,path,num_samples,verbose=True): print('Number of unique output tokens:', self.num_decoder_words) print('Max sequence length for inputs:', self.max_encoder_seq_length) print('Max sequence length for outputs:', self.max_decoder_seq_length) - - self.input_word_index = dict( - [(word, i) for i, word in enumerate(input_words)]) - self.target_word_index = dict( - [(word, i) for i, word in enumerate(target_words)]) - self.reverse_input_word_dict = dict( - (i, word) for word, i in self.input_word_index.items()) - self.reverse_target_word_dict = dict( - (i, word) for word, i in self.target_word_index.items()) + + self.input_word_index = { + word: i for i, word in enumerate(input_words)} + self.target_word_index = { + word: i for i, word in enumerate(target_words)} + self.reverse_input_word_dict = { + i: word for word, i in self.input_word_index.items()} + self.reverse_target_word_dict = { + i: word for word, i in self.target_word_index.items()} def process_input(self,input_texts,target_texts=None,verbose=True):