nlpcc2017_news_headline_categorization icon indicating copy to clipboard operation
nlpcc2017_news_headline_categorization copied to clipboard

使用预训练的词向量

Open ghost opened this issue 5 years ago • 1 comments

https://github.com/FudanNLP/nlpcc2017_news_headline_categorization/blob/067b54681f218e974c739d5a4f5ed7da26aed277/Config.py#L19

您好,为要使用我们自己预训练的词向量,并且为要确保在文本分类模型的训练过程中,词向量不参与训练。 是否需要修改该选项或其他选项?

ghost avatar Jul 31 '19 09:07 ghost

pre_trained = True 才行,不跟新参数请改下面代码:

https://github.com/FudanNLP/nlpcc2017_news_headline_categorization/blob/067b54681f218e974c739d5a4f5ed7da26aed277/model.py#L133 这行设置一个trainable=False

jingjing-gong avatar Jul 31 '19 09:07 jingjing-gong