v4 update

v3/classes/__init__.py

@@ -21,14 +21,6 @@ from .slot_dataset import SlotDataset, slot_collate
 from .papila_data import PapilaData
 from .papila_builders import build_papila_data
 from .data_bundle import DataBundle
-from .dataset import ClinicalDataset
-from .config_builder import (
-    ConfigAssembly,
-    assemble_config,
-    load_config,
-    resolve_imports,
-)
-from .filters import RegexFilter, ColumnFilter, apply_regex_filters, apply_column_filters
 from .transforms import (
     ImageTransformConfig,
     backbone_transform_config,
@@ -43,11 +35,35 @@ from .transforms import (
     TRANSFORM_REGISTRY,
     build_transform_chain,
 )
-from .model_builder import V2ModelBundle, build_model_bundle
-from .towers import ImageTower, ClinicalTower, SiameseImageTower, build_backbone
-from .bridges import Bridge, VoteBridge
-from .models import SingleEyeHT, BilateralHT
-from .v2_hypertower import V2HyperTower, V2ModeComparisonOps, V2ModeComparator
+from .towerbase import TowerBase, build_backbone, train_towers_epoch, collect_probs_towers
+from .image_towers import ImageEncoder, SiameseImageTower, ImageTower
+from .clinical_towers import ClinicalEncoder, ClinicalDataTower
+from .geometry_towers import GeometryTower
+from .hypertower_models import (
+    SingleEyeHT,
+    BilateralHT,
+    SiameseHT,
+    FusedEnsembleHT,
+    LogitMLPEnsembleHT,
+    EmbeddingMLPEnsembleHT,
+    NTowerHT,
+    NLateralHT,
+    MonoTowerHT,
+    train_single_epoch,
+    train_bilateral_epoch,
+    train_siamese_epoch,
+    train_fusion_epoch,
+    train_ntower_epoch,
+    train_mono_epoch,
+    collect_probs_classic,
+    collect_probs_ensemble,
+    collect_probs_bilateral,
+    collect_probs_siamese,
+    collect_probs_ntower,
+    collect_probs_mono,
+    V2ModeComparisonOps,
+)
+from .bridges import Bridge, HTClassifier, HyperBridge, VoteBridge
 from .hypertower_logger import HypertowerLogger
 
 __all__ = [
@@ -66,18 +82,9 @@ __all__ = [
     "PapilaData",
     "build_papila_data",
     "DataBundle",
-    "ClinicalDataset",
     "SlotLoaderFactory",
     "SlotDataset",
     "slot_collate",
-    "ConfigAssembly",
-    "assemble_config",
-    "load_config",
-    "resolve_imports",
-    "RegexFilter",
-    "ColumnFilter",
-    "apply_regex_filters",
-    "apply_column_filters",
     "ImageTransformConfig",
     "backbone_transform_config",
     "build_backbone_transform",
@@ -90,18 +97,41 @@ __all__ = [
     "UnetMaskProvider",
     "TRANSFORM_REGISTRY",
     "build_transform_chain",
-    "V2ModelBundle",
-    "build_model_bundle",
-    "ImageTower",
-    "ClinicalTower",
-    "SiameseImageTower",
+    "TowerBase",
     "build_backbone",
-    "Bridge",
-    "VoteBridge",
+    "train_towers_epoch",
+    "collect_probs_towers",
+    "ImageEncoder",
+    "SiameseImageTower",
+    "ImageTower",
+    "ClinicalEncoder",
+    "ClinicalDataTower",
+    "GeometryTower",
     "SingleEyeHT",
     "BilateralHT",
-    "V2HyperTower",
+    "SiameseHT",
+    "FusedEnsembleHT",
+    "LogitMLPEnsembleHT",
+    "EmbeddingMLPEnsembleHT",
+    "NTowerHT",
+    "NLateralHT",
+    "MonoTowerHT",
+    "train_single_epoch",
+    "train_bilateral_epoch",
+    "train_siamese_epoch",
+    "train_fusion_epoch",
+    "train_ntower_epoch",
+    "train_mono_epoch",
+    "collect_probs_classic",
+    "collect_probs_ensemble",
+    "collect_probs_bilateral",
+    "collect_probs_siamese",
+    "collect_probs_ntower",
+    "collect_probs_mono",
+    "Bridge",
+    "HTClassifier",
+    "HyperBridge",
+    "VoteBridge",
     "V2ModeComparisonOps",
-    "V2ModeComparator",
     "HypertowerLogger",
 ]