avalanche.evaluation.metrics.WandBStreamConfusionMatrix

class avalanche.evaluation.metrics.WandBStreamConfusionMatrix(class_names=None)[source]

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)[source]
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)