
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 575
使用教程
环境要求
- Python 环境
- 成对文本数据
安装与启动步骤
-
1装库
句向量训练与推理一体的包。
pip install -U angle-emb -
2先跑推理
用预训练模型编码几句,看相似度排序。
-
3准备训练数据
成对文本按格式整理。
-
4看训练器参数
命令行入口,先看清有哪些开关。
CUDA_VISIBLE_DEVICES=0 angle-trainer --help -
5导出通用格式
转成主流句向量框架格式,方便下游接。
python scripts/convert_to_sentence_transformers.py --help
如何确认成功
在自有语料上训一版后,同义句相似度高于无关句,且排序与基线一致
注意事项
- 它是训练与推理库,不是开箱检索系统。
- 训练数据质量决定上限。
- 小数据量微调容易过拟合。
- 格式转换后要复测,转换可能丢配置。
核心亮点
- 在STS和MTEB基准上达到SOTA,语义相似度效果领先
- 提供简洁API,几行代码即可完成训练和推理
- 支持多种预训练模型,灵活适配不同任务
不足之处
- 项目较新,文档和社区生态尚待完善
- 依赖较多,安装配置稍复杂
- 对中文支持需自行适配,默认资源偏英文
适用场景
- 构建语义搜索引擎,提升查询与文档匹配精度
- 文本去重和聚类,识别相似内容
- 问答系统或对话机器人中的意图识别
替代项目
sentence-transformers、SimCSE、GTE
同类项目推荐
SteptronOss
开源
一套配置跑通大模型微调、强化与评测
A lightweight, AI-native training framework for large language models. Designed for ···
VERONA
开源
一键搭建鲁棒性实验,快速算清模型抗干扰能力
A lightweight Python package for setting up robustness experiments and to compute ro···
causalml
开源
量化每个动作的真实增量,把预算花在会被打动的人身上
Uplift modeling and causal inference with machine learning algorithms
prismaquant
开源
按层敏感度自动分配量化精度,省显存不掉点
Mixed-precision quantization for LLMs. Every layer refracts into a different format ···
KDFlow
开源
一套配置跑通多种LLM蒸馏,小模型也能继承大模型能力
A user-friendly & efficient knowledge distillation framework for LLMs, supporting of···
safetensors_util
开源
轻松管理 Safetensors 模型,玩转 Stable Diffusion 与 LoRA
Utility for Safetensors Files
Firefly
开源
一键微调 20+ 主流大模型,快速定制专属 AI
Firefly: 大模型训练工具,支持训练Qwen2.5、Qwen2、Yi1.5、Phi-3、Llama3、Gemma、MiniCP···
azure-databricks-mlops-mlflow
开源
在 Azure Databricks 上快速搭建不依赖 MLflow Project 的 MLOps 流程
Azure Databricks MLOps sample for Python based source code using MLflow without usin···
focal-frequency-loss
开源
频域聚焦损失,让生成图像细节更清晰
[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis