embedding_matrix[word_dict[key