Module mls_lib.model_training.random_forest_tree_classifier_trainer

LinearRegressionTrainer: Component that trains and makes predictions.

Classes

class RandomForestTreeClassifierTrainer (features, truth, max_depth=None, n_estimators=100)

LinearRegressionTrainer: Component that trains and makes predictions.

Expand source code

class RandomForestTreeClassifierTrainer(SKLModelTrainer):
    """ LinearRegressionTrainer: Component that trains and makes predictions. """
    def __init__(self, features, truth, max_depth = None, n_estimators = 100) -> None:
        super().__init__(
            features = features,
            truth = truth,
            model = RandomForestClassifierModel(
                max_depth = max_depth,
                n_estimators = n_estimators
            )
        )

Ancestors

• SKLModelTrainer
• ModelTrainingStep
• Step

Inherited members

• SKLModelTrainer:
  • execute
  • get_output
  • is_finished
  • is_ready