- 博客(5)
- 收藏
- 关注
原创 使用Bert报错:Bert输出为字符串
在transfromer库3.X版本后,模型不再返回张量的元组,而是返回特定对象。如果版本高于4.0,那么输出的确实会是字符串,解决办法是在一开始的模型定义语句里增加一个参数return_dict=flase,让模型正确返回一个元组。如果不想考虑这么麻烦,或者加上了return_dict后,解释器报错,那就直接用x=bert_model([input_ids,attention_mask,token_type_ids])[1]即可。第一部分是所有时刻的输出,第二部分就是最后一层隐藏层的输出。
2023-12-23 10:25:24
537
1
原创 遇到问题:word2vec文本嵌入变换为使用BERT
classification.py:1143: UndefinedMetricWarning: F-score is ill-defined and being set to 0.0 in labels with no predicted sample
2023-03-23 20:46:23
146
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人