← Back to API reference neuralbench.utils.compute_class_weights_from_dataset¶ neuralbench.utils.compute_class_weights_from_dataset(train_dataset: SegmentDataset, logger: Logger, task: Literal['multiclass', 'multilabel', 'auto'] = 'auto') → tuple[dict[str, Tensor], Tensor][source][source]¶ Compute class weights from training dataset for handling class imbalance. ← Back to API reference