speaker-change-detection icon indicating copy to clipboard operation
speaker-change-detection copied to clipboard

Question for AUROC & validation step

Open Sean652039 opened this issue 7 months ago • 1 comments

Hi Housen, about the metrics part, you defined the num_classes = num_frames if scd is True, but in the initial part, the num_classes = 1 if scd is True. Could you please tell the reason behind this?

AUROC( self.num_frames if self.scd else self.num_classes, pos_label=1, average="macro", compute_on_step=False, )

And also, shall we define the task type? Since segmentation is multilable task while scd is binary task. like this task="binary" if self.scd else "multilabel",

Sean652039 avatar Jul 12 '24 20:07 Sean652039