def __init__(self, embedding: PolyEmbedding, mimic_model: Model):
self.embedding = embedding
self.mimic_model = mimic_model
self.max_word_length = K.int_shape(mimic_model.input)[1]
self.embedding_size = K.int_shape(mimic_model.output)[1]
assert len(embedding.zero_vector()) == self.embedding_size
