如何从Fastai学习器语言模型中获取句子嵌入

我已经使用Fastai成功地对语言模型进行了微调。我希望从这个微调后的模型中提取句子嵌入,以便进行句子相似度比较。我该如何获取编码器模型的嵌入?另外,这些嵌入是否可以像其他模型(如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)

我的代码如上所示,我如何从learn中获取编码?


回答:

这应该可以让你获取编码器(这是一个嵌入层):

learn.model[0].encoder

Related Posts

如何对SVC进行超参数调优?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

如何在初始训练后向模型添加训练数据?

我想在我的scikit-learn模型已经训练完成后再…

使用Google Cloud Function并行运行带有不同用户参数的相同训练作业

我正在寻找一种方法来并行运行带有不同用户参数的相同训练…

加载Keras模型,TypeError: ‘module’ object is not callable

我已经在StackOverflow上搜索并阅读了文档,…

在计算KNN填补方法中特定列中NaN值的”距离平均值”时

当我从头开始实现KNN填补方法来处理缺失数据时,我遇到…

使用巨大的S3 CSV文件或直接从预处理的关系型或NoSQL数据库获取数据的机器学习训练/测试工作

已关闭。此问题需要更多细节或更清晰的说明。目前不接受回…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注