ARFClassifier¶
Adaptive Random Forest classifier.
The 3 most important aspects of Adaptive Random Forest 1 are:
-
inducing diversity through re-sampling
-
inducing diversity through randomly selecting subsets of features for node splits
-
drift detectors per base tree, which cause selective resets in response to drifts
It also allows training background trees, which start training if a warning is detected and replace the active tree if the warning escalates to a drift.
Parameters¶
-
n_models
Type →
intDefault →
10Number of trees in the ensemble.
-
max_features (mutable)
Type →
bool | str | intDefault →
sqrtMax number of attributes for each node split.
- Ifint, then considermax_featuresat each split.
- Iffloat, thenmax_featuresis a percentage andint(max_features * n_features)features are considered per split.
- If "sqrt", thenmax_features=sqrt(n_features).
- If "log2", thenmax_features=log2(n_features).
- If None, thenmax_features=n_features. -
lambda_value (mutable)
Type →
intDefault →
6The lambda value for bagging (lambda=6 corresponds to Leveraging Bagging).
-
metric
Type →
metrics.base.MultiClassMetric | NoneDefault →
NoneMetric used to track trees performance within the ensemble. Defaults to
metrics.Accuracy(). -
disable_weighted_vote
Default →
FalseIf
True, disables the weighted vote prediction. -
drift_detector
Type →
base.DriftDetector | NoneDefault →
NoneDrift Detection method. Set to None to disable Drift detection. Defaults to
drift.ADWIN(delta=0.001). -
warning_detector
Type →
base.DriftDetector | NoneDefault →
NoneWarning Detection method. Set to None to disable warning detection. Defaults to
drift.ADWIN(delta=0.01). -
grace_period (mutable)
Type →
intDefault →
50[Tree parameter] Number of instances a leaf should observe between split attempts.
-
max_depth
Type →
int | NoneDefault →
None[Tree parameter] The maximum depth a tree can reach. If
None, the tree will grow until the system recursion limit. -
split_criterion
Type →
strDefault →
info_gain[Tree parameter] Split criterion to use.
- 'gini' - Gini
- 'info_gain' - Information Gain
- 'hellinger' - Hellinger Distance -
delta (mutable)
Type →
floatDefault →
0.01[Tree parameter] Allowed error in split decision, a value closer to 0 takes longer to decide.
-
tau (mutable)
Type →
floatDefault →
0.05[Tree parameter] Threshold below which a split will be forced to break ties.
-
leaf_prediction
Type →
strDefault →
nba[Tree parameter] Prediction mechanism used at leafs.
- 'mc' - Majority Class
- 'nb' - Naive Bayes
- 'nba' - Naive Bayes Adaptive -
nb_threshold
Type →
intDefault →
0[Tree parameter] Number of instances a leaf should observe before allowing Naive Bayes.
-
nominal_attributes
Type →
list | NoneDefault →
None[Tree parameter] List of Nominal attributes. If empty, then assume that all attributes are numerical.
-
splitter
Type →
Splitter | NoneDefault →
None[Tree parameter] The Splitter or Attribute Observer (AO) used to monitor the class statistics of numeric features and perform splits. Splitters are available in the
tree.splittermodule. Different splitters are available for classification and regression tasks. Classification and regression splitters can be distinguished by their propertyis_target_class. This is an advanced option. Special care must be taken when choosing different splitters. By default,tree.splitter.GaussianSplitteris used ifsplitterisNone. -
binary_split
Type →
boolDefault →
False[Tree parameter] If True, only allow binary splits.
-
min_branch_fraction
Type →
floatDefault →
0.01[Tree parameter] The minimum percentage of observed data required for branches resulting from split candidates. To validate a split candidate, at least two resulting branches must have a percentage of samples greater than
min_branch_fraction. This criterion prevents unnecessary splits when the majority of instances are concentrated in a single branch. -
max_share_to_split
Type →
floatDefault →
0.99[Tree parameter] Only perform a split in a leaf if the proportion of elements in the majority class is smaller than this parameter value. This parameter avoids performing splits when most of the data belongs to a single class.
-
max_size
Type →
floatDefault →
100.0[Tree parameter] Maximum memory (MiB) consumed by the tree.
-
memory_estimate_period
Type →
intDefault →
2000000[Tree parameter] Number of instances between memory consumption checks.
-
stop_mem_management
Type →
boolDefault →
False[Tree parameter] If True, stop growing as soon as memory limit is hit.
-
remove_poor_attrs
Type →
boolDefault →
False[Tree parameter] If True, disable poor attributes to reduce memory usage.
-
merit_preprune
Type →
boolDefault →
True[Tree parameter] If True, enable merit-based tree pre-pruning.
-
seed
Type →
int | NoneDefault →
NoneRandom seed for reproducibility.
Attributes¶
- models
Examples¶
from river import evaluate
from river import forest
from river import metrics
from river.datasets import synth
dataset = synth.ConceptDriftStream(
seed=42,
position=500,
width=40
).take(1000)
model = forest.ARFClassifier(seed=8, leaf_prediction="mc")
metric = metrics.Accuracy()
evaluate.progressive_val_score(dataset, model, metric)
Accuracy: 67.97%
The total number of warnings and drifts detected, respectively
model.n_warnings_detected(), model.n_drifts_detected()
(2, 1)
The number of warnings detected by tree number 2
model.n_warnings_detected(2)
1
And the corresponding number of actual concept drift detected
model.n_drifts_detected(2)
1
Methods¶
learn_one
n_drifts_detected
Get the total number of concept drifts detected, or such number on an individual tree basis (optionally).
Parameters
- tree_id —
int | None— defaults toNone
Returns
int: The number of concept drifts detected.
n_warnings_detected
Get the total number of concept drift warnings detected, or the number on an individual tree basis (optionally).
Parameters
- tree_id —
int | None— defaults toNone
Returns
int: The number of concept drift warnings detected.
predict_one
Predict the label of a set of features x.
Parameters
- x —
dict[base.typing.FeatureName, Any] - kwargs —
Any
Returns
base.typing.ClfTarget | None: The predicted label.
predict_proba_one
Predict the probability of each label for a dictionary of features x.
Parameters
- x —
dict - kwargs —
typing.Any
Returns
dict[base.typing.ClfTarget, float]: A dictionary that associates a probability which each label.
-
Heitor Murilo Gomes, Albert Bifet, Jesse Read, Jean Paul Barddal, Fabricio Enembreck, Bernhard Pfharinger, Geoff Holmes, Talel Abdessalem. Adaptive random forests for evolving data stream classification. In Machine Learning, DOI: 10.1007/s10994-017-5642-8, Springer, 2017. ↩