src.driver.fitting

Model fitting logic for MMM V2 workflow.

This module handles all aspects of model fitting including model creation, training, and loading of existing models.

Module Contents

class src.driver.fitting.ModelFitter(driver)

Handles model fitting and prediction operations for V2 architecture.

This class manages the complete model fitting workflow including model configuration, fitting, and loading of saved models. Uses the V2 architecture with in-graph scaling.

fit(model_filename: str | None = None) → src.core.base.BaseMMM

Fits a marketing model using provided data and configuration.

Uses the V2 Model architecture with in-graph scaling.

Parameters:

model_filename – Path to a saved model file (optional).

Returns:

The fitted or loaded model.

Raises:

RuntimeError – If model fitting or loading fails.