Skip to content

Add pydantic validation multidimensional MMM init #1474

Closed
@williambdean

Description

@williambdean

This can use the validate_call like with the other class

def __init__(
self,
date_column: str,
channel_columns: list[str],
target_column: str,
adstock: AdstockTransformation,
saturation: SaturationTransformation,
time_varying_intercept: bool = False,
time_varying_media: bool = False,
dims: tuple | None = None,
model_config: dict | None = None, # Ensure model_config is a dictionary
sampler_config: dict | None = None,
control_columns: list[str] | None = None,
yearly_seasonality: int | None = None,
adstock_first: bool = True,
) -> None:

Metadata

Metadata

Assignees

No one assigned

    Labels

    MMMenhancementNew feature or requestgood first issueGood for newcomers . Doesn't require extensive knowledge of the repo and package

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions