trains/logger.py
def report_confusion_matrix(self, title, series, matrix, iteration, xlabels=None, ylabels=None, comment=None):
        """
        Report a heat-map matrix

        :param title: Title (AKA metric)
        :type title: str
        :param series: Series (AKA variant)
        :type series: str
        :param matrix: A heat-map matrix (example: confusion matrix)
        :type matrix: ndarray
        :param iteration: Iteration number
        :type iteration: int
        :param xlabels: optional label per column of the matrix
        :param ylabels: optional label per row of the matrix
        :param comment: comment underneath the title
        """

        if not isinstance(matrix, np.ndarray):
            matrix = np.array(matrix)

        # if task was not started, we have to start it
        self._start_task_if_needed()

        return self._task.reporter.report_value_matrix(
            title=title,
            series=series,
            data=matrix.astype(np.float32),
            iter=iteration,
            xlabels=xlabels,
            ylabels=ylabels,
            comment=comment,
        )

        Report a heat-map matrix

        :param title: Title (AKA metric)
        :type title: str
        :param series: Series (AKA variant)
        :type series: str
        :param matrix: A heat-map matrix (example: confusion matrix)
        :type matrix: ndarray
        :param iteration: Iteration number
        :type iteration: int
        :param xlabels: optional label per column of the matrix
        :param ylabels: optional label per row of the matrix
        :param comment: comment underneath the title
        
Report a heat-map matrix
Report a heat - map matrix
def report_confusion_matrix(self, title, series, matrix, iteration, xlabels=None, ylabels=None, comment=None):
        if not isinstance(matrix, np.ndarray):
            matrix = np.array(matrix)
        self._start_task_if_needed()
        return self._task.reporter.report_value_matrix(
            title=title,
            series=series,
            data=matrix.astype(np.float32),
            iter=iteration,
            xlabels=xlabels,
            ylabels=ylabels,
            comment=comment,
        )
def report_confusion_matrix ( self , title , series , matrix , iteration , xlabels = None , ylabels = None , comment = None ) : if not isinstance ( matrix , np . ndarray ) : matrix = np . array ( matrix ) self . _start_task_if_needed ( ) return self . _task . reporter . report_value_matrix ( title = title , series = series , data = matrix . astype ( np . float32 ) , iter = iteration , xlabels = xlabels , ylabels = ylabels , comment = comment , )
Report a heat-map matrix
Report a heat - map matrix
"""
        Report a heat-map matrix

        :param title: Title (AKA metric)
        :type title: str
        :param series: Series (AKA variant)
        :type series: str
        :param matrix: A heat-map matrix (example: confusion matrix)
        :type matrix: ndarray
        :param iteration: Iteration number
        :type iteration: int
        :param xlabels: optional label per column of the matrix
        :param ylabels: optional label per row of the matrix
        :param comment: comment underneath the title
        """ # if task was not started, we have to start it
self
title
Title (AKA metric)
Title ( AKA metric )
series
Series (AKA variant)
Series ( AKA variant )
matrix
A heat-map matrix (example: confusion matrix)
A heat - map matrix ( example : confusion matrix )
iteration
None
xlabels
optional label per column of the matrix
optional label per column of the matrix
ylabels
optional label per row of the matrix
optional label per row of the matrix
comment
comment underneath the title
comment underneath the title