Module mls_lib.objects.models.random_forest_classifier_model

RandomForestClassifierModel: Component that trains and makes predictions.

Classes

class RandomForestClassifierModel (max_depth=None, n_estimators=100)

RandomForestClassifierModel: Component that trains and makes predictions.

Expand source code

class RandomForestClassifierModel(Model):
    """ RandomForestClassifierModel: Component that trains and makes predictions. """
    def __init__(self, max_depth = None, n_estimators = 100) -> None:
        super().__init__(
            model = RFC(
                max_depth = max_depth,
                n_estimators = n_estimators
            )
        )

Ancestors

• Model
• Object

Inherited members

• Model:
  • copy
  • predict
  • score
  • train