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接口啟用有條件生成（目前在筆記本中支持）。