Simplify Adaptive Random Forest (#1323)

online-ml · Sep 19, 2023 · 487341b · 487341b
1 parent 0539046
commit 487341b
Show file tree

Hide file tree

Showing 4 changed files with 193 additions and 277 deletions.
diff --git a/docs/releases/unreleased.md b/docs/releases/unreleased.md
@@ -5,3 +5,7 @@ River's mini-batch methods now support pandas v2. In particular, River conforms
 ## anomaly
 
 - Added `anomaly.LocalOutlierFactor`, which is an online version of the LOF algorithm for anomaly detection that matches the scikit-learn implementation.
+
+## forest
+
+- Simplify inner the structures of `forest.ARFClassifier` and `forest.ARFRegressor` by removing redundant class hierarchy. Simplify how concept drift logging can be accessed in individual trees and in the forest as a whole.
diff --git a/river/base/drift_detector.py b/river/base/drift_detector.py
@@ -8,7 +8,6 @@
 from __future__ import annotations
 
 import abc
-import numbers
 
 from . import base
 
@@ -58,7 +57,7 @@ class DriftDetector(_BaseDriftDetector):
     """A drift detector."""
 
     @abc.abstractmethod
-    def update(self, x: numbers.Number) -> DriftDetector:
+    def update(self, x: int | float) -> DriftDetector:
         """Update the detector with a single data point.
 
         Parameters