c
cyiing/corom-medical
模型介绍文件和版本Pull Requests讨论分析

CoROM 段落排序医疗版(nlp_corom_passage-ranking_chinese-tiny-medical)

CoROM 模型用于中文段落排序,基于医疗领域数据训练。

  • 模型 ID:iic/nlp_corom_passage-ranking_chinese-tiny-medical
  • 任务:文本排序
  • 框架:PyTorch
  • 隐藏层大小:256
  • 层数:4

验证环境

项目数值
NPUAscend910B4
CANN8.5.1
PyTorch2.9.0+cpu
torch_npu2.9.0.post1
Python3.11.14

精度评估

CPU 与 NPU 余弦相似度:

指标数值
余弦相似度1.000000

推理输出

Result: {'scores': [0.994, 0.015, 0.957]}
SUCCESS

复现

python3 inference.py
python3 evaluate.py

材料

  • inference.py - NPU 推理脚本
  • evaluate.py - CPU 与 NPU 余弦相似度评估
下载使用量0