Module mls_lib.objects.models.model

Model: Component that trains and makes predictions.

Classes

class Model (model)

Model: Component that trains and makes predictions.

Expand source code

class Model(Object):
    """ Model: Component that trains and makes predictions. """
    def __init__(self, model) -> None:
        super().__init__()
        self.model = model
    def train(self, features = None, truth = None):
        """ Train the model. """
        self.model.fit(features, truth)
    def predict(self, features = None):
        """ Make predictions. """
        return self.model.predict(features)
    def score(self, x_test = None, y_test = None):
        """ Score the model. """
        return self.model.score(x_test, y_test)

Ancestors

• Object

Subclasses

• LinearRegressionModel
• RandomForestClassifierModel
• SVMModel

Methods

def predict(self, features=None)

Make predictions.

def score(self, x_test=None, y_test=None)

Score the model.

def train(self, features=None, truth=None)

Train the model.

Inherited members

• Object:
  • copy