StructBERT Siamese UIE 模型（nlp_structbert_siamese-uie_chinese-base）

SiameseUIE 是一个基于提示词+文本构建的通用信息抽取模型，采用指针网络进行片段提取。

• 模型 ID：iic/nlp_structbert_siamese-uie_chinese-base
• 任务：Siamese UIE（通用信息抽取）
• 框架：PyTorch
• 输入：中文文本
• 输出：768 维隐藏状态

验证环境

精度评估

CPU 与 NPU 余弦相似度：

推理输出

Output shape: [1, 25, 768]
Output sample: [-0.0664, -0.0068, 0.0638, 0.0330, -0.0122]
SUCCESS

复现

python3 inference.py
python3 evaluate.py

材料

• inference.py - NPU 推理脚本
• evaluate.py - CPU 与 NPU 余弦相似度评估