Fix class imbalance in CE loss#56

Merged

michaelmckinsey1 merged 11 commits intoLBANN:mainfrom

PatrickRMiles:miles30/class_imbalance

May 6, 2026

Commits on May 6, 2026

use class weights in CE loss to make background less dominant; calc weights at trainer init
Patrick Miles
committed
ruff
Patrick Miles
committed
missing import
Patrick Miles
committed
ruff
Patrick Miles
committed
ruff
Patrick Miles
committed
fix missing volume dtype
Patrick Miles
committed
remove default ce_weight_num_samples

PatrickRMiles
and
michaelmckinsey1
authored
remove default ce_weight_num_samples in trainer

PatrickRMiles
and
michaelmckinsey1
authored
move ce loss helpers to losses.py
Patrick Miles
committed
sample by fraction of total rather than hard number
Patrick Miles
committed
ruff
Patrick Miles
committed