feat(Stacking): Implemented ensemble learning technique Stacking

frankkramer-lab · May 22, 2022 · 168116f · 168116f
1 parent 511154b
commit 168116f
Show file tree

Hide file tree

Showing 2 changed files with 550 additions and 1 deletion.
diff --git a/aucmedi/ensemble/__init__.py b/aucmedi/ensemble/__init__.py
@@ -31,7 +31,7 @@
 | ------------------------------------------ | ---------------------------------------------------------------------------------------------------- |
 | [Augmenting][aucmedi.ensemble.augmenting]  | Inference Augmenting (test-time augmentation) function for augmenting unknown images for prediction. |
 | [Bagging][aucmedi.ensemble.bagging]        | Cross-Validation based Bagging for equal models trained with different sampling.                     |
-| Stacking                                   | Coming soon. |
+| [stacking][aucmedi.ensemble.stacking]      | Ensemble of unequal models with a fitted Metalearner stacked on top of it. |
 
 !!! info
     ![EnsembleLearning_overview](../../images/ensemble.theory.png)
@@ -48,3 +48,4 @@
 #-----------------------------------------------------#
 from aucmedi.ensemble.augmenting import predict_augmenting
 from aucmedi.ensemble.bagging import Bagging
+from aucmedi.ensemble.stacking import Stacking