sklearn

SciKitModel(model, output_name)

Bases: Model

A model that wraps a scikit-learn model.

Initialize the model.

Parameters:

Name	Type	Description	Default
model	Any	The scikit-learn model.	required
output_name	str	The name of the output column.	required

Source code in src/flowcean/models/sklearn.py

def __init__(
    self,
    model: Any,
    output_name: str,
) -> None:
    """Initialize the model.

    Args:
        model: The scikit-learn model.
        output_name: The name of the output column.
    """
    self.model = model
    self.output_name = output_name