面向水库群联合调度的小样本命名实体识别方法

本发明公开了一种面向水库群联合调度的小样本命名实体识别方法，通过实体跨度检测和实体类型分类两个阶段构建模型。在跨度检测阶段使用跨度边界矩阵，学习跨度边界信息并过滤潜在错误跨度，提高检测精度；在实体类型分类阶段，通过标签引导提高原型网络构建质量。在模型训练过程中，使用对比学习改进损失函数，通过反向传播更新参数，使相同类型实体的空间表示更加接近。完成实体原型构造后，利用距离函数计算样本与类别原型之间的距离，并将其转化为相似度概率对样本进行分类。本发明实现了水库群联合调度实体的自动提取，能够在样本较少的数据集中完成命名实体识别任务，具有较高的识别效果，极大节省了人工标记所需的人力物力与时间成本。