crowncode-backend / app /training /train_deep_classifiers.py

Commit History

fix: update evaluation to use optimal threshold and adjust BCEWithLogitsLoss for class imbalance
9b6c85d

Rthur2003 commited on

fix: adjust BCEWithLogitsLoss to handle class imbalance with pos_weight
5ac76c0

Rthur2003 commited on

fix: add missing roc_curve metric import for evaluation
bfbdc0b

Rthur2003 commited on

feat: add TorchSklearnWrapper for PyTorch model compatibility with sklearn and enhance final model training process
bb33480

Rthur2003 commited on

feat: add deep learning classifier comparison script for AURIS with multiple architectures
b8d143b

Rthur2003 commited on