基于BERT的中文陈述类型分类

基于 BERT 的中文陈述类型三分类模型。输入一个陈述句，判断其类型为 FACT（事实）、OPINION（观点）还是 OTHER（其他）。

输入数据应该为如下结构

{
    "statement_text": "我觉得这个方案不错。"
}

• statement_text：待判断类型的陈述句

模型输出为一个概率分布和预测标签，具体而言，被组装为如下格式

{
  "predicted_label": "OPINION",
  "probabilities": {
    "FACT": 0.0009,
    "OPINION": 0.9982,
    "OTHER": 0.0001
  }
}

• predicted_label：预测的陈述类型标签
• probabilities：三类概率分布

可以在下载本项目后用如下代码测试模型效果

python main.py