tdlm_model.py 文件源码

def generate_on_topic(self, sess, topic_id, start_word_id, temperature=1.0, max_length=30, stop_word_id=None): 
        if topic_id != -1:
            topic_emb = sess.run(tf.expand_dims(tf.nn.embedding_lookup(self.topic_output_embedding, topic_id), 0))
        else:
            topic_emb = None
        return self.generate(sess, topic_emb, start_word_id, temperature, max_length, stop_word_id)

    #generate a sequence of words, given a document