问题描述
我已经为文本分类任务微调了一个 bert 基础模型。现在,我想提取隐藏层输出,以便将此输出与其他特征结合起来训练随机森林模型。 问题在于我不知道如何提取隐藏层输出。如果有人能在这方面帮助我,那就太好了。
from simpletransformers.classification import ClassificationModel
model_xlm = ClassificationModel('bert','bert-base-uncased')
model_xlm.train_model(df_train)
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)