Documentation
搜索文档…
Databricks
W&B通过在Databricks环境中定制W&B Jupyter笔记本体验,与Databricks进行集成。

Databricks配置

在集群中安装wandb

导航到你的集群配置,选择你的集群,点击库,然后在安装新(Install New)上,选择PyPI并添加包wandb

认证

为了认证你的W&B账户,你可以添加一个你的笔记本可以查询的datatricks secret。
1
# install databricks clipip install databricks-cli​# Generate a token from databricks UIdatabricks configure --token​# Create a scope with one of the two commands (depending if you have security features enabled on databricks):# with security add-ondatabricks secrets create-scope --scope wandb# without security add-ondatabricks secrets create-scope --scope wandb --initial-manage-principal users​# Add your api_key from: https://app.wandb.ai/authorizedatabricks secrets put --scope wandb --key api_key
Copied!

示例

简单

1
import osimport wandb​api_key = dbutils.secrets.get("wandb", "api_key")wandb.login(key=api_key)​wandb.init()wandb.log({"foo": 1})
Copied!

扫描(Sweep)

试图使用wandb.sweep() 或 wandb.agent()的笔记本需要(临时)设置。
1
import os# These will not be necessary in the futureos.environ['WANDB_ENTITY'] = "my-entity"os.environ['WANDB_PROJECT'] = "my-project-that-exists"
Copied!
我们在这里介绍了更多关于如何在笔记本中运行扫描(sweep)的细节:从Jupyter 笔记本上扫描(sweep)/sweeps/python-api
最近更新 9mo ago