概述
小波AI提供了Python SDK
,提供了平台包括数据集、特征库、模型训练、模型注册等一系列AI核心功能。扩大了AI-Dev的适用场景
适用场景
Python SDK
实现了大量平台端的核心功能,仅需要少量代码就可以完成平台数据的交互工作,而不需要进入网站操作。所以SDK可以广泛适用于:
- 结合平台
定时任务
模块的功能,使用SDK可以周期性自动完成数据预处理任务 - 可以通过SDK获取权限内的数据集数据,用于离线的
数据分析
、特征工程
等大量线下的数据调试工作 - 结合SDK的日志记录和追踪功能,可广泛用于
模型建模
、离线训练
等场景,通过SDK自动记录实验数据和结果到平台,便于算法工程师快速比对实验结果。同时也便于管理人员了解相关工作进度,促进更好的协作 - 可以直接通过SDK,完成
模型离线注册
的工作,熟练操作后将会比网页操作更节省时间,进一步提升工作效率
SDK安装
使用pip
命令安装SDK
pip install waveletai
SDK使用
使用api_token参数登录
import waveletai
waveletai.init(api_token="YOUR_TOKEN")
登录成功后,控制台会输出如下信息为登录成功:
WaveletAI Backend connected
登录后即可使用SDK所有功能了,具体参考各个模块接口说明
SDK核心功能
账号同步
Python SDK
关联平台账号,一键登录账号,支持多端同时登录,自动获取账号相关所有业务数据权限
数据管理
dataset
模块,与网站功能类似,SDK提供了创建数据集、上传、下载数据等功能。大文件支持自动分段上传,下载
模型管理
app
、model
模块,与网站功能类似,SDK提供了创建、获取应用、模型等功能
离线训练
使用experiment
模块,SDK提供了创建模型试验、获取实验
注册模型库
使用model
-ModelVersion
,可以注册本地模型到平台模型库,用于后续的模型发布等服务