Skip to main content
Version: 2.17.1

概述

小波AI提供了Python SDK,提供了平台包括数据集、特征库、模型训练、模型注册等一系列AI核心功能。扩大了AI-Dev的适用场景

适用场景

Python SDK实现了大量平台端的核心功能,仅需要少量代码就可以完成平台数据的交互工作,而不需要进入网站操作。所以SDK可以广泛适用于:

  1. 结合平台定时任务模块的功能,使用SDK可以周期性自动完成数据预处理任务
  2. 可以通过SDK获取权限内的数据集数据,用于离线的数据分析特征工程等大量线下的数据调试工作
  3. 结合SDK的日志记录和追踪功能,可广泛用于模型建模离线训练等场景,通过SDK自动记录实验数据和结果到平台,便于算法工程师快速比对实验结果。同时也便于管理人员了解相关工作进度,促进更好的协作
  4. 可以直接通过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提供了创建数据集、上传、下载数据等功能。大文件支持自动分段上传,下载

模型管理

appmodel模块,与网站功能类似,SDK提供了创建、获取应用、模型等功能

离线训练

使用experiment模块,SDK提供了创建模型试验、获取实验

注册模型库

使用model-ModelVersion,可以注册本地模型到平台模型库,用于后续的模型发布等服务