pub struct RandomForestRegressor<F> {
pub n_estimators: usize,
pub max_depth: Option<usize>,
pub max_features: MaxFeatures,
pub min_samples_split: usize,
pub min_samples_leaf: usize,
pub random_state: Option<u64>,
/* private fields */
}Expand description
Random forest regressor.
Builds an ensemble of decision tree regressors, each trained on a bootstrap sample with a random subset of features considered at each split. Final predictions are the mean across all trees.
§Type Parameters
F: The floating-point type (f32orf64).
Fields§
§n_estimators: usizeNumber of trees in the forest.
max_depth: Option<usize>Maximum depth of each tree. None means unlimited.
max_features: MaxFeaturesStrategy for the number of features considered at each split.
min_samples_split: usizeMinimum number of samples required to split an internal node.
min_samples_leaf: usizeMinimum number of samples required in a leaf node.
random_state: Option<u64>Random seed for reproducibility. None means non-deterministic.
Implementations§
Source§impl<F: Float> RandomForestRegressor<F>
impl<F: Float> RandomForestRegressor<F>
Sourcepub fn new() -> Self
pub fn new() -> Self
Create a new RandomForestRegressor with default settings.
Defaults: n_estimators = 100, max_depth = None,
max_features = All, min_samples_split = 2,
min_samples_leaf = 1, random_state = None.
Sourcepub fn with_n_estimators(self, n_estimators: usize) -> Self
pub fn with_n_estimators(self, n_estimators: usize) -> Self
Set the number of trees.
Sourcepub fn with_max_depth(self, max_depth: Option<usize>) -> Self
pub fn with_max_depth(self, max_depth: Option<usize>) -> Self
Set the maximum tree depth.
Sourcepub fn with_max_features(self, max_features: MaxFeatures) -> Self
pub fn with_max_features(self, max_features: MaxFeatures) -> Self
Set the maximum features strategy.
Sourcepub fn with_min_samples_split(self, min_samples_split: usize) -> Self
pub fn with_min_samples_split(self, min_samples_split: usize) -> Self
Set the minimum number of samples to split a node.
Sourcepub fn with_min_samples_leaf(self, min_samples_leaf: usize) -> Self
pub fn with_min_samples_leaf(self, min_samples_leaf: usize) -> Self
Set the minimum number of samples in a leaf.
Sourcepub fn with_random_state(self, seed: u64) -> Self
pub fn with_random_state(self, seed: u64) -> Self
Set the random seed for reproducibility.
Trait Implementations§
Source§impl<F: Clone> Clone for RandomForestRegressor<F>
impl<F: Clone> Clone for RandomForestRegressor<F>
Source§fn clone(&self) -> RandomForestRegressor<F>
fn clone(&self) -> RandomForestRegressor<F>
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
source. Read moreSource§impl<F: Debug> Debug for RandomForestRegressor<F>
impl<F: Debug> Debug for RandomForestRegressor<F>
Source§impl<F: Float> Default for RandomForestRegressor<F>
impl<F: Float> Default for RandomForestRegressor<F>
Source§impl<'de, F> Deserialize<'de> for RandomForestRegressor<F>
impl<'de, F> Deserialize<'de> for RandomForestRegressor<F>
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Source§impl<F: Float + Send + Sync + 'static> Fit<ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>> for RandomForestRegressor<F>
impl<F: Float + Send + Sync + 'static> Fit<ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>> for RandomForestRegressor<F>
Source§fn fit(
&self,
x: &Array2<F>,
y: &Array1<F>,
) -> Result<FittedRandomForestRegressor<F>, FerroError>
fn fit( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<FittedRandomForestRegressor<F>, FerroError>
Fit the random forest regressor.
§Errors
Returns FerroError::ShapeMismatch if x and y have different
numbers of samples.
Returns FerroError::InsufficientSamples if there are no samples.
Returns FerroError::InvalidParameter if n_estimators is 0.
Source§type Fitted = FittedRandomForestRegressor<F>
type Fitted = FittedRandomForestRegressor<F>
fit.Source§type Error = FerroError
type Error = FerroError
fit.Source§impl<F: Float + Send + Sync + 'static> PipelineEstimator<F> for RandomForestRegressor<F>
impl<F: Float + Send + Sync + 'static> PipelineEstimator<F> for RandomForestRegressor<F>
Source§fn fit_pipeline(
&self,
x: &Array2<F>,
y: &Array1<F>,
) -> Result<Box<dyn FittedPipelineEstimator<F>>, FerroError>
fn fit_pipeline( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<Box<dyn FittedPipelineEstimator<F>>, FerroError>
Auto Trait Implementations§
impl<F> Freeze for RandomForestRegressor<F>
impl<F> RefUnwindSafe for RandomForestRegressor<F>where
F: RefUnwindSafe,
impl<F> Send for RandomForestRegressor<F>where
F: Send,
impl<F> Sync for RandomForestRegressor<F>where
F: Sync,
impl<F> Unpin for RandomForestRegressor<F>where
F: Unpin,
impl<F> UnsafeUnpin for RandomForestRegressor<F>
impl<F> UnwindSafe for RandomForestRegressor<F>where
F: UnwindSafe,
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> DistributionExt for Twhere
T: ?Sized,
impl<T> DistributionExt for Twhere
T: ?Sized,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more