- 博客(5)
- 收藏
- 关注
RSS订阅原创 使用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
393
原创 遇到问题：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
98
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人