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)