WANDBResultTracker¶

class WANDBResultTracker(project, experiment=None, offline=False, **kwargs)[source]¶

Bases: pykeen.trackers.base.ResultTracker

A tracker for Weights and Biases.

Note that you have to perform wandb login beforehand.

Initialize result tracking via WANDB.

Parameters

project (str) – project name your WANDB login has access to.
experiment (Optional[str]) – The experiment name to appear on the website. If not given, WANDB will generate a random name.

Methods Summary

`log_metrics`(metrics[, step, prefix])	Log metrics to result store.
`log_params`(params[, prefix])	Log parameters to result store.

Methods Documentation

log_metrics(metrics, step=None, prefix=None)[source]¶

Log metrics to result store.

Parameters

metrics (Mapping[str, float]) – The metrics to log.
step (Optional[int]) – An optional step to attach the metrics to (e.g. the epoch).
prefix (Optional[str]) – An optional prefix to prepend to every key in metrics.

Return type

None

log_params(params, prefix=None)[source]¶

Log parameters to result store.