nlpcc2017_news_headline_categorization
nlpcc2017_news_headline_categorization copied to clipboard
使用预训练的词向量
https://github.com/FudanNLP/nlpcc2017_news_headline_categorization/blob/067b54681f218e974c739d5a4f5ed7da26aed277/Config.py#L19
您好,为要使用我们自己预训练的词向量,并且为要确保在文本分类模型的训练过程中,词向量不参与训练。 是否需要修改该选项或其他选项?
pre_trained = True
才行,不跟新参数请改下面代码:
https://github.com/FudanNLP/nlpcc2017_news_headline_categorization/blob/067b54681f218e974c739d5a4f5ed7da26aed277/model.py#L133 这行设置一个trainable=False