trw.callbacks.callback_export_classification_report

Module Contents

Classes

CallbackExportClassificationReport

Export the main classification measures for the classification outputs of the model

Functions

get_mappinginv(datasets_infos, dataset_name, split_name, class_name)

Extract the mappinginv from the datasets_infos

Attributes

logger

trw.callbacks.callback_export_classification_report.get_mappinginv(datasets_infos, dataset_name, split_name, class_name)

Extract the mappinginv from the datasets_infos

trw.callbacks.callback_export_classification_report.logger

class trw.callbacks.callback_export_classification_report.CallbackExportClassificationReport(with_confusion_matrix=True, with_ROC=True, with_report=True)

Bases: trw.callbacks.callback.Callback

Export the main classification measures for the classification outputs of the model

This include: * text report (e.g., accuracy, sensitivity, specificity, F1, typical errors & confusion matrix) * confusion matrix plot * ROC & AUC for binary classification problems

max_class_names = 40

__call__(self, options, history, model, losses, outputs, datasets, datasets_infos, callbacks_per_batch, **kwargs)