Repolode
apple

apple/embedding-atlas

用于交互式探索大规模 embedding 数据及其元数据的可视化工具,支持交叉过滤与搜索。

Star
4,990
Fork
332
未关闭 Issue
34
最近提交
6小时前

这是什么?

Embedding Atlas 是一个用于可视化大规模 embedding 及其元数据的工具,可以查看数据的整体结构、密集区域和离群点,并按查询内容或已有数据点搜索相似样本。

针对 embedding 数据,它提供带标签的自动聚类、核密度估计与密度等高线、与绘制顺序无关的透明度渲染,以及近邻搜索。渲染基于 WebGPU,面向最多数百万个点的数据集。对于任意表格数据,它支持联动仪表盘和交叉过滤,内置柱状图、折线图、气泡图、计数图和 eCDF 图表,并可通过可组合的图表配置自定义热力图等图形;文本、图像、音频、数值、分类和时间列均有对应的查看器。

项目可以作为 Python 命令行工具使用,也可以作为 Jupyter 等 notebook 中的组件,或以 npm 包形式在 React、Svelte 中引入。它还提供 MCP 接口,供 AI 智能体查询数据结构、执行 SQL 并创建图表。

核心功能

  • 01对 embedding 点进行自动聚类并添加标签
  • 02核密度估计与密度等高线显示
  • 03实时搜索与近邻查询
  • 04基于 WebGPU 渲染,可处理数百万级别的点
  • 05图表之间联动并支持跨列交叉过滤
  • 06通过 MCP 让 AI 智能体执行 SQL 并创建图表

适用场景

  • 探索文本或图像 embedding 数据集的结构与离群样本
  • 通过查找已知样本的近邻来检查检索效果
  • 在 notebook 或 Web 应用中嵌入交互式 embedding 查看器
  • 让 AI 智能体分析并绘制表格数据图表

快速上手

  1. 1

    安装 Python 包

    pip install embedding-atlas
  2. 2

    在你的数据集上启动该工具

    embedding-atlas <your-dataset>

使用前须知

  • 需要支持 WebGPU 的浏览器才能获得流畅的渲染性能。

标签

常见问题

embedding-atlas 是什么?
用于交互式探索大规模 embedding 数据及其元数据的可视化工具,支持交叉过滤与搜索。 Embedding Atlas 是一个用于可视化大规模 embedding 及其元数据的工具,可以查看数据的整体结构、密集区域和离群点,并按查询内容或已有数据点搜索相似样本。
embedding-atlas 可以免费使用吗?用的什么许可证?
embedding-atlas 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
embedding-atlas 还在维护吗?
活跃维护中。embedding-atlas 最近一次提交是在 2026年10月21日。
embedding-atlas 适合用来做什么?
探索文本或图像 embedding 数据集的结构与离群样本 通过查找已知样本的近邻来检查检索效果 在 notebook 或 Web 应用中嵌入交互式 embedding 查看器 让 AI 智能体分析并绘制表格数据图表
embedding-atlas 怎么安装和运行?
安装 Python 包: pip install embedding-atlas 在你的数据集上启动该工具: embedding-atlas <your-dataset>
在哪里获取 embedding-atlas?
源代码和安装说明在 GitHub 上:https://github.com/apple/embedding-atlas

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目