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