src/__init__.py

# ACL-LKNet: Anatomically Aware Self-Supervised Large-Kernel Network
# for ACL Tear Detection in Knee MRI

from .config import Config
from .dataset import MRNetDataset, MRNetSSLDataset, create_dataloaders, get_stratified_folds
from .evaluate import (
    compute_metrics,
    compute_metrics_with_ci,
    find_optimal_thresholds,
    generate_threshold_sweep,
    plot_threshold_curves,
    plot_confusion_matrices_dual,
    evaluate_slice_explainability,
    evaluate_perturbation_faithfulness,
    evaluate_slice_localization,
    evaluate_msm_reconstruction,
    evaluate_scanner_perturbation_robustness,
    delong_test,
    mcnemar_test,
    full_evaluation,
)

try:
    from .models.acl_lknet import ACLLKNet, create_model_from_config
    from .models.msm import MaskedSliceModeling
    from .train import train_supervised, pretrain_msm, train_5fold_cross_validation
except ImportError:
    pass