AnglE

用角度学习训练最强句子向量,登顶MTEB榜单

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 搜索知识 🍴 复刻 37 🐛 待处理 14 👀 关注 7
GitHub ↗
GitHub 星标 ★ 575
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

AnglE 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类搜索知识
开发团队SeanLee97
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型dense-retrieval,embeddings,information-retrieval,llama,llama2,llm,mteb,rag,retrieval-augmented-generation,semantic-similarity,semantic-textual-similarity,sentence-embedding,sentence-embeddings,sentence-vector,sts,stsbenchmark,text-embedding,text-similarity,text-vector,text2vec
GitHub 星标★ 575
上线时间2023-10-17 00:00:00
代码更新2026-03-22
仓库体积1.3 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数7

项目介绍

AnglE是一个用于训练和推理高质量句子嵌入(Sentence Embeddings)的Python开源库,旨在解决文本语义相似度计算、检索和聚类等任务中嵌入表示能力不足的问题。它实现了最新的AnglE(Angle Embedding)方法,通过引入角度空间中的对比学习,显著提升了嵌入的判别力。项目在STS(语义文本相似度)基准和MTEB(大规模文本嵌入基准)排行榜上达到SOTA水平,支持多种预训练模型(如BERT、RoBERTa等)的微调,并提供了简洁的API用于快速训练和推理。核心能力包括:高效的训练脚本、灵活的模型配置、丰富的评估工具,以及易于集成的推理接口。AnglE特别适合需要高精度语义匹配的场景,如语义搜索、文本去重、问答系统等。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 575

使用教程

难度:进阶 约 35 分钟 部署方式:本地安装 5 步

环境要求

  • Python 环境
  • 成对文本数据

安装与启动步骤

  1. 1装库

    句向量训练与推理一体的包。

    pip install -U angle-emb
  2. 2先跑推理

    用预训练模型编码几句,看相似度排序。

  3. 3准备训练数据

    成对文本按格式整理。

  4. 4看训练器参数

    命令行入口,先看清有哪些开关。

    CUDA_VISIBLE_DEVICES=0 angle-trainer --help
  5. 5导出通用格式

    转成主流句向量框架格式,方便下游接。

    python scripts/convert_to_sentence_transformers.py --help

如何确认成功

在自有语料上训一版后,同义句相似度高于无关句,且排序与基线一致

注意事项

  • 它是训练与推理库,不是开箱检索系统。
  • 训练数据质量决定上限。
  • 小数据量微调容易过拟合。
  • 格式转换后要复测,转换可能丢配置。

核心亮点

  • 在STS和MTEB基准上达到SOTA,语义相似度效果领先
  • 提供简洁API,几行代码即可完成训练和推理
  • 支持多种预训练模型,灵活适配不同任务

不足之处

  • 项目较新,文档和社区生态尚待完善
  • 依赖较多,安装配置稍复杂
  • 对中文支持需自行适配,默认资源偏英文

适用场景

  • 构建语义搜索引擎,提升查询与文档匹配精度
  • 文本去重和聚类,识别相似内容
  • 问答系统或对话机器人中的意图识别

替代项目

sentence-transformers、SimCSE、GTE

上一篇:RAG-FiT

下一篇:DataMate

同类项目推荐

SteptronOss 开源

一套配置跑通大模型微调、强化与评测

A lightweight, AI-native training framework for large language models. Designed for ···

★ 589 2026-09-15
VERONA 开源

一键搭建鲁棒性实验,快速算清模型抗干扰能力

A lightweight Python package for setting up robustness experiments and to compute ro···

★ 52 2026-08-09
causalml 开源

量化每个动作的真实增量,把预算花在会被打动的人身上

Uplift modeling and causal inference with machine learning algorithms

★ 6043 2026-10-25
prismaquant 开源

按层敏感度自动分配量化精度,省显存不掉点

Mixed-precision quantization for LLMs. Every layer refracts into a different format ···

★ 105 2026-10-31
KDFlow 开源

一套配置跑通多种LLM蒸馏,小模型也能继承大模型能力

A user-friendly & efficient knowledge distillation framework for LLMs, supporting of···

★ 256 2026-10-07
safetensors_util 开源

轻松管理 Safetensors 模型,玩转 Stable Diffusion 与 LoRA

Utility for Safetensors Files

★ 102 2026-08-09
Firefly 开源

一键微调 20+ 主流大模型,快速定制专属 AI

Firefly: 大模型训练工具,支持训练Qwen2.5、Qwen2、Yi1.5、Phi-3、Llama3、Gemma、MiniCP···

★ 6641 2026-08-09
azure-databricks-mlops-mlflow 开源

在 Azure Databricks 上快速搭建不依赖 MLflow Project 的 MLOps 流程

Azure Databricks MLOps sample for Python based source code using MLflow without usin···

★ 97 2026-08-09
focal-frequency-loss 开源

频域聚焦损失,让生成图像细节更清晰

[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis

★ 716 2026-08-09