Downloads · 30 days
6
24% of all-time downloads
SunnyAxe/bert_country_infer
bert_country_infer is a text classification model from SunnyAxe. Use it when you need a label for a piece of text. It is set up for transformers.
本模型是在SRTP项目中,为中文文学领域文献的国别板块推理任务而训练的基于BERT的模型。
Downloads · 30 days
6
24% of all-time downloads
All-time downloads
25
Public
Parameters
102M
409 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors409 MB · 100%
From the Hugging Face model README
本模型是在SRTP项目中,为中文文学领域文献的国别板块推理任务而训练的基于BERT的模型。
input文本最大长度150,label个数11。
label与推理出的数值对应如下:
{0: '比较文学', 1: '大洋洲', 2: '东欧、北欧', 3: '翻译研究', 4: '非洲', 5: '加拿大及其他美洲国家', 6: '美国', 7: '文艺理论与批评', 8: '西欧', 9: '亚洲', 10: '中欧、南欧'}
input: 论文标题+论文关键词+论文摘要(拼接) output: 板块(0~10)