Module mls_lib.model_training.linear_regression_trainer

LinearRegressionTrainer: Component that trains and makes predictions.

Classes

class LinearRegressionTrainer (features, truth)

LinearRegressionTrainer: Component that trains and makes predictions.

Expand source code

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

Ancestors

• SKLModelTrainer
• ModelTrainingStep
• Step

Inherited members

• SKLModelTrainer:
  • execute
  • get_output
  • is_finished
  • is_ready