xujiashuai/nlp_corom_sentence-embedding_chinese-tiny-ecom
模型介绍文件和版本Pull Requests讨论分析

CoROM 句向量模型(nlp_corom_sentence-embedding_chinese-tiny-ecom)

CoROM 是一款基于 BERT 的轻量级句向量模型,专为电商领域优化。

  • 模型 ID:iic/nlp_corom_sentence-embedding_chinese-tiny-ecom
  • 任务:句向量生成
  • 框架:PyTorch
  • 输入:中文文本
  • 输出:256 维句向量

验证环境

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

精度评估

CPU 与 NPU 余弦相似度:

指标数值
余弦相似度1.000000

推理输出

Embedding shape: [1, 256]
Embedding sample: [0.1568, -1.2536, 0.1713, -0.5272, -0.1829]
SUCCESS

复现

python3 inference.py
python3 evaluate.py

材料

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