fix: adding tests for new ensembles

adaptive-machine-learning · May 30, 2024 · 7ecd80d · 7ecd80d
1 parent 25f0f75
commit 7ecd80d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/test_classifiers.py b/tests/test_classifiers.py
@@ -15,7 +15,9 @@
     HoeffdingAdaptiveTree,
     SAMkNN,
     DynamicWeightedMajority,
-    CSMOTE
+    CSMOTE,
+    LeveragingBagging,
+    OnlineAdwinBagging,
 )
 from capymoa.base import Classifier
 from capymoa.base import MOAClassifier
@@ -58,6 +60,8 @@
         (partial(SAMkNN), 82.65, 82.0, None),
         (partial(DynamicWeightedMajority), 84.05, 89.0, None),
         (partial(CSMOTE), 80.55, 79.0, None),
+        (partial(LeveragingBagging), 86.7, 91.0, None),
+        (partial(OnlineAdwinBagging), 85.25, 92.0, None),
     ],
     ids=[
         "OnlineBagging",
@@ -79,6 +83,8 @@
         "SAMkNN",
         "DynamicWeightedMajority",
         "CSMOTE",
+        "LeveragingBagging",
+        "OnlineAdwinBagging",
     ],
 )
 def test_classifiers(