trainer

Gretel培训师

该模块旨在提供一个简单的接口，以帮助用户在具有高行和列计数的复杂数据集上成功训练合成模型，并提供基于云SaaS的培训和基于多GPU的并行化等功能。使用Gretel.ai的API键免费入门。

当前功能和功能：

• 具有以下功能的文本，表格和时间序列数据的合成数据生成器：
  • 使用条件数据生成来平衡数据集或增强少数族裔类。
  • 自动数据验证。
  • 合成数据质量报告。
  • 隐私过滤器和可选差异隐私支持。
• 支持多种模型类型：
  • Gretel-LSTM模型类型支持文本，表格，时间序列和条件数据生成。
  • Gretel-ACTGAN模型类型支持表格和条件数据生成。
  • Gretel-GPT自然语言综合基于GPT-3的开源实施（即将推出）。
  • Gretel-DGAN多变量时间序列基于Doppelganger（即将推出）。

现在尝试一下！

如果您想快速开始使用gretel.ai合成数据，只需单击下面的按钮并按照示例。请参阅./notebooks文件夹中的其他Python3和Jupyter笔记本示例。

加入综合数据社区不和谐

如果您想成为合成数据社区的一部分，以接收最新版本的公告，提出问题，建议新功能或参加开发会议，请加入合成数据社区服务器！

安装

使用pip ：

pip install -U gretel-trainer

Quickstart

1。通过Gretel CLI添加Gretel API键。

使用Gretel客户端将API密钥存储到磁盘上。此步骤是可选的，培训师将在下一步中提示API键。

gretel configure

2。使用Gretel API进行训练或微调模型

 from gretel_trainer import trainer

dataset = "https://gretel-public-website.s3-us-west-2.amazonaws.com/datasets/USAdultIncome5k.csv"

model = trainer . Trainer ()
model . train ( dataset )

3。生成合成数据！

 df = model . generate ()

发展

设置环境和安装依赖项。

python -m venv venv
source venv/bin/activate
pip install -r requirements-dev.txt
pip install -e .

• 通过make test
• 运行类型检查（有限的覆盖范围）通过make type

托多斯 /路线图

• 通过SDK接口启用有条件生成（目前在笔记本中支持）。