从 SimpleTransformer 的分类模型中提取层输出

问题描述

我已经为文本分类任务微调了一个 bert 基础模型。现在,我想提取隐藏层输出,以便将此输出与其他特征结合起来训练随机森林模型。 问题在于我不知道如何提取隐藏层输出。如果有人能在这方面帮助我,那就太好了。

from simpletransformers.classification import ClassificationModel

model_xlm = ClassificationModel('bert','bert-base-uncased')
model_xlm.train_model(df_train)

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)