WANDBResultTracker

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

Bases: pykeen.trackers.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 (Dict[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.

Return type

None