src/proteingnn/model.py
def _log_pr_curve(self, y, pred):
        """Log PR curve in classification."""
        assert not self._regression
        tensorboard = self.logger.experiment
        for class_i in range(self.num_classes):
            tensorboard.add_pr_curve(f'PR_curve_class_{class_i}', pred[:, class_i], y[:, class_i],
                                     global_step=self.global_step)
Log PR curve in classification.
Log PR curve in classification.
Log PR curve in classification .
def _log_pr_curve(self, y, pred):
        assert not self._regression
        tensorboard = self.logger.experiment
        for class_i in range(self.num_classes):
            tensorboard.add_pr_curve(f'PR_curve_class_{class_i}', pred[:, class_i], y[:, class_i],
                                     global_step=self.global_step)
def _log_pr_curve ( self , y , pred ) : assert not self . _regression tensorboard = self . logger . experiment for class_i in range ( self . num_classes ) : tensorboard . add_pr_curve ( f'PR_curve_class_{class_i}' , pred [ : , class_i ] , y [ : , class_i ] , global_step = self . global_step )
Log PR curve in classification.
Log PR curve in classification .
"""Log PR curve in classification."""
self
y
pred