基岩平台客户端库
bdrk的Python项目详细描述
Bedrock帮助数据科学家拥有机器学习工作流的端到端部署。bdrk
是在基岩平台上与api交互的官方客户端库。
用法
为了使用bdrk
,您需要向basis ai注册一个帐户。请通过电子邮件contact@basis-ai.com
开始。创建帐户后,将向您颁发一个可用于向基岩进行身份验证的个人api令牌。
安装基岩客户端
可以使用以下命令从pypi安装基岩客户机库。我们建议在虚拟环境中运行它,以防止潜在的依赖冲突。
pip install bdrk
注意,python 3.7及更高版本正式支持客户端库。
安装可选依赖项
可以安装以下可选依赖项以启用其他功能。
命令行支持:
pip install bdrk[cli]
Feauture商店支持:
pip install bdrk[fs]
设置您的环境
安装后,需要在项目的根目录中添加格式良好的bedrock.hcl
配置文件。配置文件指定要运行哪个脚本进行培训和部署,以及它们各自的基本Docker映像。您可以找到一个示例目录布局here。
在本地使用模块时,可能需要为bdrk
定义以下环境变量,以便对基岩进行api调用。当在集群中运行时,这些变量将在工作负载容器上自动设置。
BEDROCK_API_DOMAIN=https://bedrock.basis-ai.com BEDROCK_API_TOKEN=<your personal API token>
记录培训指标
使用bdrk
基岩客户机,您可以通过向train.py
添加日志代码,轻松地将培训度量导出到基岩。下面的示例演示了在基岩平台上可视化的日志图表和度量。
importloggingfrombedrock_client.bedrock.apiimportBedrockApilogger=logging.getLogger(__name__)bedrock=BedrockApi(logger)bedrock.log_metric("Accuracy",0.97)bedrock.log_chart_data([0,1,2,3],[0.1,0.5,0.7,0.9])