SDV

SDV

使用机器学习生成高质量合成数据,提高隐私保护

SDV利用多种机器学习算法,提供生成表格合成数据的解决方案。主要功能包括生成单表、多表和序列数据,支持数据预处理、匿名化和逻辑约束定义。此外,SDV还提供数据评估和可视化工具,比较合成数据与真实数据,并生成质量报告。适合初学者和资深数据科学家,满足多样化需求。

Synthetic Data VaultSDVPython机器学习数据生成Github开源项目
<div align="center"> <br/> <p align="center"> <i>此仓库是 <a href="https://sdv.dev">The Synthetic Data Vault Project</a> 的一部分,该项目由 <a href="https://datacebo.com">DataCebo</a> 提供支持。</i> </p>

开发状态 PyPi节 单元测试 集成测试 覆盖状况 下载 Colab Slack

<div align="left"> <br/> <p align="center"> <a href="https://github.com/sdv-dev/SDV"> <img align="center" width=40% src="https://yellow-cdn.veclightyear.com/35dd4d3f/f735a4eb-e1e4-4f50-8564-4aef8aed43f8.png"></img> </a> </p> </div> </div>

概述

Synthetic Data Vault (SDV) 是一个Python库,旨在成为您的合成表格数据创建的一站式解决方案。SDV使用多种机器学习算法来学习真实数据中的模式,并在合成数据中模仿它们。

特性

:brain: 使用机器学习生成合成数据。 SDV提供了多种模型,从经典的统计方法(GaussianCopula)到深度学习方法(CTGAN)均可应用。可以为单个表、多张互相关联的表或序列表生成数据。

:bar_chart: 评估和可视化数据。 根据各种度量标准,将合成数据与实际数据进行比较。诊断问题并生成质量报告以获得更多见解。

:arrows_counterclockwise: 预处理、匿名化和定义约束。 控制数据处理以提高合成数据的质量,从不同类型的匿名化方法中选择,并以逻辑约束的形式定义业务规则。

重要链接
教程通过SDV获得一些动手体验。启动教程笔记本并亲自运行代码。
:book: [文档]通过用户指南和API参考来学习如何使用SDV库。
:orange_book: [博客]获取更多关于使用SDV、部署模型以及我们的合成数据社区的见解。
社区加入我们的Slack工作区获取公告和讨论。
:computer: [网站]查看SDV网站以获取有关项目的更多信息。

安装

SDV是根据商业源代码许可证公开提供的。使用pip或conda安装SDV。我们建议使用虚拟环境以避免与设备上其他软件发生冲突。

pip install sdv
conda install -c pytorch -c conda-forge sdv

入门

加载一个演示数据集以开始。此数据集是描述虚构酒店客人住宿的单张表。

from sdv.datasets.demo import download_demo real_data, metadata = download_demo( modality='single_table', dataset_name='fake_hotel_guests')

单表元数据示例

演示还包括元数据,它是对数据集的描述,包括每列中的数据类型和主键(guest_email)。

合成数据

接下来,我们可以创建一个SDV合成器,这是一个可以用来创建合成数据的对象。它从真实数据中学习模式,并复制这些模式生成合成数据。让我们使用GaussianCopulaSynthesizer

from sdv.single_table import GaussianCopulaSynthesizer synthesizer = GaussianCopulaSynthesizer(metadata) synthesizer.fit(data=real_data)

现在合成器已准备好生成合成数据!

synthetic_data = synthesizer.sample(num_rows=500)

合成数据将具有以下特性:

  • 敏感列完全匿名化。 邮件、账单地址和信用卡号列包含新数据,因此您不会暴露真实值。
  • 其他列遵循统计模式。 例如,房间类型的比例、入住日期的分布和房价与房间类型之间的相关性得到了保留。
  • 键和其他关系保持完整。 主键(客人邮箱)对每行都是唯一的。如果您有多个表,主键和外键之间的连接是有意义的。

评估合成数据

SDV库允许您通过将合成数据与实际数据进行比较来评估合成数据。首先生成一个质量报告。

from sdv.evaluation.single_table import evaluate_quality quality_report = evaluate_quality( real_data, synthetic_data, metadata)
生成报告中...

(1/2) 评估列形状: |████████████████| 9/9 [00:00<00:00, 1133.09it/s]|
列形状得分: 89.11%

(2/2) 评估列对趋势: |██████████████████████████████████████████| 36/36 [00:00<00:00, 502.88it/s]|
列对趋势得分: 88.3%

总体得分(平均值):88.7%

该对象计算了一个0到100%(100为最佳)的总体质量得分,并给出详细的分解。为了获得更多见解,您还可以将合成数据与真实数据进行可视化比较。

from sdv.evaluation.single_table import get_column_plot fig = get_column_plot( real_data=real_data, synthetic_data=synthetic_data, column_name='amenities_fee', metadata=metadata ) fig.show()

真实数据与合成数据

接下来做什么?

使用SDV库,您可以合成单表、多表和序列数据。您还可以定制完整的合成数据工作流,包括预处理、匿名化和添加约束。

要了解更多信息,请访问SDV演示页面

致谢

感谢我们多年来建设和维护SDV生态系统的团队!

查看贡献者

引用

如果您在研究中使用了SDV,请引用以下论文:

Neha Patki, Roy Wedge, Kalyan Veeramachaneni. The Synthetic Data Vault. IEEE DSAA 2016.

@inproceedings{
    SDV,
    title={The Synthetic data vault},
    author={Patki, Neha and Wedge, Roy and Veeramachaneni, Kalyan},
    booktitle={IEEE International Conference on Data Science and Advanced Analytics (DSAA)},
    year={2016},
    pages={399-410},
    doi={10.1109/DSAA.2016.49},
    month={Oct}
}

<div align="center"> <a href="https://datacebo.com"><picture> <source media="(prefers-color-scheme: dark)" srcset="https://github.com/sdv-dev/SDV/blob/stable/docs/images/datacebo-logo-dark-mode.png"> <img align="center" width=40% src="https://yellow-cdn.veclightyear.com/35dd4d3f/f2b4dcf5-818f-4574-baff-3f73f12834e0.png"></img> </picture></a> </div> <br/> <br/>

The Synthetic Data Vault Project 于2016年在MIT的数据到AI实验室首次创建。经过4年的研究并在企业中获得了足够的发展,于2020年成立了DataCebo,旨在推动这一项目的发展。 今天,DataCebo自豪地成为SDV的开发者,SDV是合成数据生成和评估的最大生态系统。它是多个支持合成数据的库的家园,包括:

  • 🔄 数据发现和转换。反转转换以再现现实数据。
  • 🧠 多种机器学习模型——从Copulas到深度学习——用于创建表格、多表和时间序列数据。
  • 📊 测量合成数据的质量和隐私性,并比较不同的合成数据生成模型。

开始使用SDV包——一个完整的集成解决方案和您合成数据的一站式选择。或者,使用特定需求的独立库。

编辑推荐精选

讯飞智文

讯飞智文

一键生成PPT和Word,让学习生活更轻松

讯飞智文是一个利用 AI 技术的项目,能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定,还是学生群体的职业生涯规划、实习避坑指南,亦或是活动策划、旅游攻略等内容,它都能提供支持,帮助用户精准表达,轻松呈现各种信息。

AI办公办公工具AI工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图热门
讯飞星火

讯飞星火

深度推理能力全新升级,全面对标OpenAI o1

科大讯飞的星火大模型,支持语言理解、知识问答和文本创作等多功能,适用于多种文件和业务场景,提升办公和日常生活的效率。讯飞星火是一个提供丰富智能服务的平台,涵盖科技资讯、图像创作、写作辅助、编程解答、科研文献解读等功能,能为不同需求的用户提供便捷高效的帮助,助力用户轻松获取信息、解决问题,满足多样化使用场景。

热门AI开发模型训练AI工具讯飞星火大模型智能问答内容创作多语种支持智慧生活
Spark-TTS

Spark-TTS

一种基于大语言模型的高效单流解耦语音令牌文本到语音合成模型

Spark-TTS 是一个基于 PyTorch 的开源文本到语音合成项目,由多个知名机构联合参与。该项目提供了高效的 LLM(大语言模型)驱动的语音合成方案,支持语音克隆和语音创建功能,可通过命令行界面(CLI)和 Web UI 两种方式使用。用户可以根据需求调整语音的性别、音高、速度等参数,生成高质量的语音。该项目适用于多种场景,如有声读物制作、智能语音助手开发等。

Trae

Trae

字节跳动发布的AI编程神器IDE

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

AI工具TraeAI IDE协作生产力转型热门
咔片PPT

咔片PPT

AI助力,做PPT更简单!

咔片是一款轻量化在线演示设计工具,借助 AI 技术,实现从内容生成到智能设计的一站式 PPT 制作服务。支持多种文档格式导入生成 PPT,提供海量模板、智能美化、素材替换等功能,适用于销售、教师、学生等各类人群,能高效制作出高品质 PPT,满足不同场景演示需求。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

热门AI辅助写作AI工具讯飞绘文内容运营AI创作个性化文章多平台分发AI助手
材料星

材料星

专业的AI公文写作平台,公文写作神器

AI 材料星,专业的 AI 公文写作辅助平台,为体制内工作人员提供高效的公文写作解决方案。拥有海量公文文库、9 大核心 AI 功能,支持 30 + 文稿类型生成,助力快速完成领导讲话、工作总结、述职报告等材料,提升办公效率,是体制打工人的得力写作神器。

openai-agents-python

openai-agents-python

OpenAI Agents SDK,助力开发者便捷使用 OpenAI 相关功能。

openai-agents-python 是 OpenAI 推出的一款强大 Python SDK,它为开发者提供了与 OpenAI 模型交互的高效工具,支持工具调用、结果处理、追踪等功能,涵盖多种应用场景,如研究助手、财务研究等,能显著提升开发效率,让开发者更轻松地利用 OpenAI 的技术优势。

Hunyuan3D-2

Hunyuan3D-2

高分辨率纹理 3D 资产生成

Hunyuan3D-2 是腾讯开发的用于 3D 资产生成的强大工具,支持从文本描述、单张图片或多视角图片生成 3D 模型,具备快速形状生成能力,可生成带纹理的高质量 3D 模型,适用于多个领域,为 3D 创作提供了高效解决方案。

3FS

3FS

一个具备存储、管理和客户端操作等多种功能的分布式文件系统相关项目。

3FS 是一个功能强大的分布式文件系统项目,涵盖了存储引擎、元数据管理、客户端工具等多个模块。它支持多种文件操作,如创建文件和目录、设置布局等,同时具备高效的事件循环、节点选择和协程池管理等特性。适用于需要大规模数据存储和管理的场景,能够提高系统的性能和可靠性,是分布式存储领域的优质解决方案。

下拉加载更多