avalanche.evaluation.metrics.WandBStreamConfusionMatrix

class avalanche.evaluation.metrics.WandBStreamConfusionMatrix(class_names=None)

Confusion Matrix metric compatible with Weights and Biases logger. Differently from the StreamConfusionMatrix, this metric will use W&B built-in functionalities to log the Confusion Matrix.

This metric may not produce meaningful outputs with other loggers.

https://docs.wandb.ai/guides/track/log#custom-charts

__init__(class_names=None)

Parameters:

class_names – list of names for the classes. E.g. [“cat”, “dog”] if class 0 == “cat” and class 1 == “dog” If None, no class names will be used. Default None.

Methods

__init__([class_names])	param class_names: list of names for the classes.
after_backward(strategy)
after_eval(strategy)
after_eval_dataset_adaptation(strategy)
after_eval_exp(strategy)
after_eval_forward(strategy)
after_eval_iteration(strategy)
after_forward(strategy)
after_train_dataset_adaptation(strategy)
after_training(strategy)
after_training_epoch(strategy)
after_training_exp(strategy)
after_training_iteration(strategy)
after_update(strategy)
before_backward(strategy)
before_eval(strategy)
before_eval_dataset_adaptation(strategy)
before_eval_exp(strategy)
before_eval_forward(strategy)
before_eval_iteration(strategy)
before_forward(strategy)
before_train_dataset_adaptation(strategy)
before_training(strategy)
before_training_epoch(strategy)
before_training_exp(strategy)
before_training_iteration(strategy)
before_update(strategy)
reset()	Resets the metric internal state.
result()	Obtains the value of the metric.
update(output, target)