« Back to API reference neuralbench.callbacks.PlotConfusionMatrix¶ class neuralbench.callbacks.PlotConfusionMatrix(labels: list[str] | None = None)[source][source]¶ on_test_epoch_end(trainer, pl_module)[source][source]¶ Called when the test epoch ends. on_validation_epoch_end(trainer, pl_module)[source][source]¶ Called when the val epoch ends. « Back to API reference