From cef8c86cd21ef0c306dde4460a82c30641d40f32 Mon Sep 17 00:00:00 2001 From: Carlos Trujillo <59846724+carlangastr@users.noreply.github.com> Date: Fri, 29 Sep 2023 23:17:51 +0000 Subject: [PATCH] Adding Experimental note --- pymc_marketing/mmm/base.py | 6 +++--- pyproject.toml | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pymc_marketing/mmm/base.py b/pymc_marketing/mmm/base.py index 0e6fd2a6..7df781df 100644 --- a/pymc_marketing/mmm/base.py +++ b/pymc_marketing/mmm/base.py @@ -617,7 +617,7 @@ def plot_budget_scenearios( self, *, base_data: Dict, method: str, **kwargs ) -> plt.Figure: """ - Plots the budget and contribution bars side by side for multiple scenarios. + Experimental: Plots the budget and contribution bars side by side for multiple scenarios. Parameters ---------- @@ -842,7 +842,7 @@ def optimize_channel_budget_for_maximum_contribution( parameters: Dict[str, Tuple[float, float]], ) -> pd.DataFrame: """ - Optimize the allocation of a given total budget across multiple channels to maximize the expected contribution. + Experimental: Optimize the allocation of a given total budget across multiple channels to maximize the expected contribution. The optimization is based on the method provided, where each channel's contribution follows a saturating function of its allocated budget. The function seeks the budget allocation @@ -896,7 +896,7 @@ def compute_channel_curve_optimization_parameters_original_scale( self, method: str ) -> Dict: """ - Estimate the parameters for the saturating function of each channel's contribution. + Experimental: Estimate the parameters for the saturating function of each channel's contribution. The function estimates the parameters (alpha, constant) for each channel based on the specified method (either 'sigmoid' or 'michaelis-menten'). These parameters represent the maximum possible contribution (alpha) and the constant parameter which vary their definition based on the function (constant) for each channel. diff --git a/pyproject.toml b/pyproject.toml index fce29b4e..14461308 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,7 +14,7 @@ maintainers = [{ name = "PyMC Labs", email = "info@pymc-labs.io" }] dependencies = [ "arviz>=0.13.0", "matplotlib>=3.5.1", - "numpy>=1.17,<1.25", + "numpy>=1.17", "pandas", # NOTE: Keep minimum pymc version in sync with ci.yml `OLDEST_PYMC_VERSION` "pymc>=5.8.2",