1

我能够使用快速人工智能微调语言模型。我想从用于句子相似性的微调模型中提取句子嵌入。如何获得编码器模型嵌入?嵌入也可以与其他模型(如 USE)中的其他嵌入进行比较吗?

data_lm = TextLMDataBunch.from_df(train_df = se1, valid_df = se2, path = "",text_cols='text')
learn = language_model_learner(data_lm,drop_mult=0.7,pretrained=True,arch=AWD_LSTM)
learn.fit_one_cycle(3, 1e-01)

我的代码在上面我如何从学习中获取编码?

4

1 回答 1

1

这应该给你编码器(这是一个嵌入层):

learn.model[0].encoder

于 2019-08-19T15:44:34.720 回答