google-meridian 1.3.1__py3-none-any.whl → 1.3.2__py3-none-any.whl

{google_meridian-1.3.1.dist-info → google_meridian-1.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-meridian
-Version: 1.3.1
+Version: 1.3.2
 Summary: Google's open source mixed marketing model library, helps you understand your return on investment and direct your ad spend with confidence. 
 Author-email: The Meridian Authors <no-reply@google.com>
 Project-URL: homepage, https://github.com/google/meridian
@@ -25,9 +25,9 @@ Requires-Dist: pandas<3,>=2.2.2
 Requires-Dist: patsy<1,>=0.5.3
 Requires-Dist: scipy<2,>=1.13.1
 Requires-Dist: statsmodels>=0.14.5
-Requires-Dist: tensorflow<2.19,>=2.18
+Requires-Dist: tensorflow<2.21,>=2.18
 Requires-Dist: tensorflow-probability<0.26,>=0.25
-Requires-Dist: tf-keras<2.19,>=2.18
+Requires-Dist: tf-keras<2.21,>=2.18
 Requires-Dist: xarray
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
@@ -38,12 +38,12 @@ Provides-Extra: colab
 Requires-Dist: psutil; extra == "colab"
 Requires-Dist: python-calamine; extra == "colab"
 Provides-Extra: and-cuda
-Requires-Dist: tensorflow[and-cuda]<2.19,>=2.18; extra == "and-cuda"
+Requires-Dist: tensorflow[and-cuda]<2.21,>=2.18; extra == "and-cuda"
 Provides-Extra: mlflow
 Requires-Dist: mlflow; extra == "mlflow"
 Provides-Extra: jax
-Requires-Dist: jax==0.4.26; extra == "jax"
-Requires-Dist: jaxlib==0.4.26; extra == "jax"
+Requires-Dist: jax==0.5.3; extra == "jax"
+Requires-Dist: jaxlib==0.5.3; extra == "jax"
 Requires-Dist: tensorflow-probability[substrates-jax]==0.25.0; extra == "jax"
 Provides-Extra: schema
 Requires-Dist: mmm-proto-schema; extra == "schema"
@@ -203,7 +203,7 @@ To cite this repository:
   author = {Google Meridian Marketing Mix Modeling Team},
   title = {Meridian: Marketing Mix Modeling},
   url = {https://github.com/google/meridian},
-  version = {1.3.1},
+  version = {1.3.2},
   year = {2025},
 }
 ```

{google_meridian-1.3.1.dist-info → google_meridian-1.3.2.dist-info}/RECORD

@@ -1,39 +1,29 @@
-google_meridian-1.3.1.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+google_meridian-1.3.2.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
 meridian/__init__.py,sha256=0fOT5oNZF7-pbiWWGUefV-ysafttieG079m1ijMFQO8,861
-meridian/constants.py,sha256=ZmMIoJDFQvKIOVG9oPOQ7Cj16wt4HDS5fCPBrz_KiLE,20308
-meridian/version.py,sha256=mmCkrGRWB8mI33apVnverT2ysfJofSyhlXSCVNotj9U,644
-meridian/analysis/__init__.py,sha256=NWDhtRkKs-n66C6746rXT7Wk8tLdkrT9NDvuPG_B_5c,874
-meridian/analysis/analyzer.py,sha256=EUKoN69EEz4vt4BpyRvrwEoHt7eLZ8UM6emNe14Tbj0,219975
-meridian/analysis/formatter.py,sha256=AN2M4jdUV88XjNdi5-dK_mITES1J1Dk3Zs7DYo3OTKg,7290
-meridian/analysis/optimizer.py,sha256=RX8HXSUi-tvLSjzF2OIbXQT3pMT75PzA5Y0xK6PPE_I,126277
-meridian/analysis/summarizer.py,sha256=7hHBf2Bj0lKqamg7DGvaKIuCf06CSsPPhDM2wKsXUPk,19035
+meridian/constants.py,sha256=idvYuDmoULgHvVG5kGJb2j2VAdBF6BeOwDfKLP14-Fo,20322
+meridian/version.py,sha256=eMNUh0AywCHr0ZyrDyJchcddVsAyWWKKHuXI0VxLScE,644
+meridian/analysis/__init__.py,sha256=AM7xpqoeC-mmY4tPIyHisjQ2MICI7v3jSri--DhDqXA,874
+meridian/analysis/analyzer.py,sha256=qGKYoEpWC6pDf2sDrwLhpa2putDnpUzbBpuNqL1lGi4,219910
+meridian/analysis/optimizer.py,sha256=6OqJg8T09g4GqJc6_DnYDJsBAlY4ToUb3MRtS0EH4Nc,126316
+meridian/analysis/summarizer.py,sha256=pah0osYYJFQTf2Fn7-nnvxlBupf9cfaKW87bcqRO1A4,19036
 meridian/analysis/summary_text.py,sha256=I_smDkZJYp2j77ea-9AIbgeraDa7-qUYyb-IthP2qO4,12438
 meridian/analysis/test_utils.py,sha256=aZq88pxtpMHwhcpfYz8nHhR0Alhi_OvgS9qBR4LBgO0,78346
-meridian/analysis/visualizer.py,sha256=1vjX3XQF_BoXEGDUS4o2GheZQ5c0VGdHuc-nXySnpsY,94091
+meridian/analysis/visualizer.py,sha256=_z5vSnXyv3Zx4tmKLyevnMolzLzUzNFcNInOipEmMok,94092
 meridian/analysis/review/__init__.py,sha256=cF24EbhiVSs-tvtRf59uVin39tu6aCTTCaeEdv6ISZ8,804
-meridian/analysis/review/checks.py,sha256=CBBZ0uNN91b_0H9DGdbDRVdiTRI4D7xptLCq_gK5WAI,25002
+meridian/analysis/review/checks.py,sha256=QEUVwC8L9Pif3y0B_OVAUpZeN6EnFGKH2oXA_dsdgbc,26893
 meridian/analysis/review/configs.py,sha256=5JJ8v6n22GNBmE78xNX6jwdjkZz2qar4Q9YTcVqzcoI,3653
-meridian/analysis/review/constants.py,sha256=bL-se2BFfRxPr3F0Rfpdy7d8s3idMoaka_t6hN1K3Sc,1369
-meridian/analysis/review/results.py,sha256=jsOPohmwozCrx8yt9p7_GiVxGmRDVeAbA8W48MUfAPA,16168
+meridian/analysis/review/constants.py,sha256=9tnnc_vAaJi9mnZ0GrRr86RsVq5fyhBaEtIvOLNmn8A,1498
+meridian/analysis/review/results.py,sha256=ZZiAdFrqySzcbjrCEacBLQS_ddiklZ34yD1BWns5SYI,17418
 meridian/analysis/review/reviewer.py,sha256=BcfmqHjp-30iZBlrzWfXDN1IJU-UIjINxZ7lsrj5Mts,6675
-meridian/analysis/templates/card.html.jinja,sha256=pv4MVbQ25CcvtZY-LH7bFW0OSeHobkeEkAleB1sfQ14,1284
-meridian/analysis/templates/chart.html.jinja,sha256=87i0xnXHRBoLLxBpKv2i960TLToWq4r1aVQZqaXIeMQ,1086
-meridian/analysis/templates/chips.html.jinja,sha256=t-ovn-2pcPCn1ev1VtSaSC3W7QxSy_iAO_L5-p68mdg,1030
-meridian/analysis/templates/insights.html.jinja,sha256=6hEWipbOMiMzs9QGZ6dcB_73tNkj0ZtNiC8E89a98zg,606
-meridian/analysis/templates/stats.html.jinja,sha256=9hQOG02FX1IHVIvdWS_-LI2bbSaqdyHEtCZkiArwAg0,772
-meridian/analysis/templates/style.css,sha256=RODTWc2pXcG9zW3q9SEJpVXgeD-WwQgzLpmFcbXPhLg,5492
-meridian/analysis/templates/style.scss,sha256=nSrZOpcIrVyiL4eC9jLUlxIZtAKZ0Rt8pwfk4H1nMrs,5076
-meridian/analysis/templates/summary.html.jinja,sha256=LuENVDHYIpNo4pzloYaCR2K9XN1Ow6_9oQOcOwD9nGg,1707
-meridian/analysis/templates/table.html.jinja,sha256=mvLMZx92RcD2JAS2w2eZtfYG-6WdfwYVo7pM8TbHp4g,1176
-meridian/backend/__init__.py,sha256=ftXcdb3tIky_m8exhD9RRhaEqTEvMcqZ8lGkkR1PjsE,39495
+meridian/backend/__init__.py,sha256=i2XOZnIPo3xAKpMGT39-4CVnVdf4cJyO0jB9FEE4zlQ,41337
 meridian/backend/config.py,sha256=B9VQnhBfg9RW04GNbt7F5uCugByenoJzt-keFLLYEp8,3561
-meridian/backend/test_utils.py,sha256=DYU5IpWiyM26aTY9Q84mOGRw0dQ9XmsZRsAJNFuZDp0,11667
+meridian/backend/test_utils.py,sha256=oJNosF_x_BzNuia8LzLFb_YfjGWHRCzR5FXNN5KQ8sw,13738
 meridian/data/__init__.py,sha256=StIe-wfYnnbfUbKtZHwnAQcRQUS8XCZk_PCaEzw90Ww,929
 meridian/data/arg_builder.py,sha256=Kqlt88bOqFj6D3xNwvWo4MBwNwcDFHzd-wMfEOmLoPU,3741
 meridian/data/data_frame_input_data_builder.py,sha256=_hexZMFAuAowgo6FaOGElHSFHqhGnHQwEEBcwnT3zUE,27295
 meridian/data/input_data.py,sha256=Qlxm4El6h1SRPsWDqZoKkOcMtrjiRWr3z8sU2mtghRA,43151
 meridian/data/input_data_builder.py,sha256=tbZjVXPDfmtndVyJA0fmzGzZwZb0RCEjXOTXb-ga8Nc,25648
-meridian/data/load.py,sha256=X2nmYCC-7A0RUgmdolTqCt0TD3NEZabQ5oGv-TugE00,40129
+meridian/data/load.py,sha256=ETX8Z62Gk6JcxFxvyB4XQhpNcRSqBRIO4_sTAN58mCY,40172
 meridian/data/nd_array_input_data_builder.py,sha256=lfpmnENGuSGKyUd7bDGAwoLqHqteOKmHdKl0VI2wCQA,16341
 meridian/data/test_utils.py,sha256=mw-QPTP15oXf32I7cxMe8iSFBLB3seqEiITZMTz_Eg8,59838
 meridian/data/time_coordinates.py,sha256=C5A5fscSLjPH6G9YT8OspgIlCrkMY7y8dMFEt3tNSnE,9874
@@ -41,7 +31,7 @@ meridian/mlflow/__init__.py,sha256=elwXUqPQYi7VF9PYjelU1tydfcUrmtuoq6eJCOnV9bk,6
 meridian/mlflow/autolog.py,sha256=SZsrynLjozcyrAFCNWiqchSa2yOszVnwFBGz23BmWUU,6379
 meridian/model/__init__.py,sha256=mhF5VkRxvwamRa_0AihgbFuXLMueRCK-Je_ZZvU5IFw,1013
 meridian/model/adstock_hill.py,sha256=HoRKjyL03pCTBz6Utof9wEvlQCFM43BvrEW_oupj7NU,17688
-meridian/model/knots.py,sha256=87kw5oa3T1k9GgT_aWXTqQx5XCxLsS2w1hnzc581XL0,26677
+meridian/model/knots.py,sha256=B78oTQ97Zd0aON4CnhMPqJZ4eamy6d-esKMWqoDf9uQ,27273
 meridian/model/media.py,sha256=skjy4Vd8LfDQWlqR_2lJ1qbG9UcS1dow5W45BAu4qk8,14599
 meridian/model/model.py,sha256=jMtfl7woWtJ8M8AX42QeZ5hUS8hlhPdZ-9OU8KahjKA,68984
 meridian/model/model_test_data.py,sha256=XGBz8RGdCsjAUOmgxX3CfWSj-_hdq2Lc8saFCqmImwM,23901
@@ -50,13 +40,23 @@ meridian/model/prior_distribution.py,sha256=ZArW4uXIPPQL6hRWiGZUzcHktbkjE_vOklvl
 meridian/model/prior_sampler.py,sha256=iLvCefhA4WY0ENcnLK9471WUZPPyzQ1je58MRjxKv74,25460
 meridian/model/spec.py,sha256=VlK6WJiPo2lzOF0O2judtJ6O3uEw7wYL5AT8bioq4gE,19188
 meridian/model/transformers.py,sha256=HxlVJitxP-wu-NOHU0tArFUZ4NAO3c7adAYj4Zvqnvo,8363
-meridian/model/eda/__init__.py,sha256=w3p7ZUZLq5TOEHm8n2P1CWjGrzuNrkqSSnVFdlw17Dk,812
-meridian/model/eda/constants.py,sha256=V9aOHQDvB3WAEyT0NE4gE8rqbStaGVh3XDlBPOKpuLc,739
-meridian/model/eda/eda_engine.py,sha256=5Ikgiz-6d3uTBan71WwsgPHOqk0fAy390ZLCq9L6HoY,64846
-meridian/model/eda/eda_outcome.py,sha256=P-0kNIbNXcyqMaNvFxiL3x6fhtYOL2trw-zPPZGXh5w,5670
+meridian/model/eda/__init__.py,sha256=bMj9kd2LWU_LQZAjQv54FFggzdv4CKRYblvc-0cHXc4,768
+meridian/model/eda/constants.py,sha256=Kt8x8hvC2WkeEtW0Wmid8GkqZbh_p6NdiZ_A5V3qzwM,1031
+meridian/model/eda/eda_engine.py,sha256=AxvKxdH8Q_TWlGZ58bWcfGMSgxkHZ2wrHloPuzS5C4Y,73324
+meridian/model/eda/eda_outcome.py,sha256=cR9M49e6bDrBNxHOThW3aQlX5gZCOENC7GBljKQx7OY,6475
 meridian/model/eda/eda_spec.py,sha256=diieYyZH0ee3ZLy0rGFMcWrrgiUrz2HctMwOrmtJR6w,2871
-meridian/model/eda/meridian_eda.py,sha256=GTdBaAtfsHS5s6P5ZESaeh1ElKV_o7dSqQosiMnFBKg,7537
-schema/__init__.py,sha256=Df2XKjMKa0ry7CTlDuJnyxuTdSgAd5za922wMlE98cg,681
+meridian/templates/card.html.jinja,sha256=gaQSHh8xN-LqHzI-xsHaORsaeOD7hnxkq-Fcw-ytZ8M,1284
+meridian/templates/chart.html.jinja,sha256=j8rNLOfXxKx4f5TpP9_W8uCM8zKys8ga75jv-XCny2g,1086
+meridian/templates/chips.html.jinja,sha256=IhPmAI7qZkXj5NeLUKVMCSKFzwv9yR8Qq140t36U10A,1030
+meridian/templates/formatter.py,sha256=75KtZA2y6iyxpkVM3CLmpUtJ1-umBP9xXxdXkkSZcr8,7547
+meridian/templates/formatter_test.py,sha256=nlefFTQaSArNyZWzGUuzCKtt3P3PJNZFmkGm0XMYwBY,8543
+meridian/templates/insights.html.jinja,sha256=IRm49X49HZX-rc6z0uZmpsA_1JLD83UhI2BacOanHLY,606
+meridian/templates/stats.html.jinja,sha256=JMRC_W2PyReU30ISOd8uv2ugDpYmSU3XtPk2OTsaQus,772
+meridian/templates/style.css,sha256=O7YCKVqXgK4Ms3nnGmCVW4hILn4GCBRe4e15XJaP4ww,5492
+meridian/templates/style.scss,sha256=he5jXpGatNfI6vtNbqdENapLiwIlXjqIBkKwxNVEvyw,5076
+meridian/templates/summary.html.jinja,sha256=kR4nQc-oGBwXN6buKcf_wPaoWASomR4fT4gJ1iD9hYw,1775
+meridian/templates/table.html.jinja,sha256=ibgnjdKCjS2qbq61lxqwtlb7JigDrc2m42RDdQvQnKQ,1176
+schema/__init__.py,sha256=grEfsjrcPRzhd89ZXLggnBnrryl97bsayKWP1ue1ffE,1206
 schema/serde/__init__.py,sha256=xyydIcWB5IUpcn3wu1m9HL1fK4gMWURbwTyRsQtolF0,975
 schema/serde/constants.py,sha256=aYtD_RuA0GCkpC4TIQq3VjMqEc837Wn-TlJNm-yn_4Y,1842
 schema/serde/distribution.py,sha256=jy3h6JD1TSs4gwociMis814sz_Fm2kFQ2UbkgjYJW9k,19347
@@ -70,7 +70,7 @@ schema/serde/serde.py,sha256=8vUqhJxvZgX9UY3rXTyWJznRgapwDzzaHXDHwV_kKTA,1612
 schema/serde/test_data.py,sha256=7hfEWyvZ9WcAkVAOXt6elX8stJlsfhfd-ASlHo9SRb8,107342
 schema/utils/__init__.py,sha256=AkC4NMbmXC3PFBY9dFYxlf3qFsxt5OOBVdc9zmFXsC8,675
 schema/utils/time_record.py,sha256=-KzHFjvSBUUXsfESPAfcJP_VFxaFLqj90Ac0kgKWfpI,4624
-google_meridian-1.3.1.dist-info/METADATA,sha256=wu5D6r6v46vd5g4uKLfXEVuhxkO3jwEgS68wn6m0jR4,9547
-google_meridian-1.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-google_meridian-1.3.1.dist-info/top_level.txt,sha256=yWkWDLV_UUanhKmk_xNPiKNdPDl1oyU1sBYwEnhaSf4,16
-google_meridian-1.3.1.dist-info/RECORD,,
+google_meridian-1.3.2.dist-info/METADATA,sha256=BHbc_4zpZ5EBv1urcvQR_brE9WEJaia4F1ZLYaA8sG0,9545
+google_meridian-1.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+google_meridian-1.3.2.dist-info/top_level.txt,sha256=yWkWDLV_UUanhKmk_xNPiKNdPDl1oyU1sBYwEnhaSf4,16
+google_meridian-1.3.2.dist-info/RECORD,,

meridian/analysis/__init__.py

@@ -15,9 +15,8 @@
 """Meridian analysis API for trained models."""
 
 from meridian.analysis import analyzer
-from meridian.analysis import formatter
 from meridian.analysis import optimizer
 from meridian.analysis import review
 from meridian.analysis import summarizer
 from meridian.analysis import visualizer
-
+from meridian.templates import formatter

meridian/analysis/analyzer.py

@@ -53,7 +53,6 @@ def _validate_non_media_baseline_values_numbers(
       )
 
 
-# TODO: Refactor the related unit tests to be under DataTensors.
 @dataclasses.dataclass
 class DataTensors(backend.ExtensionType):
   """Container for data variable arguments of Analyzer methods.

meridian/analysis/optimizer.py

@@ -27,10 +27,10 @@ import jinja2
 from meridian import backend
 from meridian import constants as c
 from meridian.analysis import analyzer as analyzer_module
-from meridian.analysis import formatter
 from meridian.analysis import summary_text
 from meridian.data import time_coordinates as tc
 from meridian.model import model
+from meridian.templates import formatter
 import numpy as np
 import pandas as pd
 import xarray as xr
@@ -1174,10 +1174,12 @@ class OptimizationResults:
       diff = self.optimized_data.total_cpik - self.nonoptimized_data.total_cpik
       non_optimized_performance_title = summary_text.NON_OPTIMIZED_CPIK_LABEL
       non_optimized_performance_stat = (
-          f'${self.nonoptimized_data.total_cpik:.2f}'
+          f'{currency}{self.nonoptimized_data.total_cpik:.2f}'
       )
       optimized_performance_title = summary_text.OPTIMIZED_CPIK_LABEL
-      optimized_performance_stat = f'${self.optimized_data.total_cpik:.2f}'
+      optimized_performance_stat = (
+          f'{currency}{self.optimized_data.total_cpik:.2f}'
+      )
       optimized_performance_diff = formatter.compact_number(diff, 2, currency)
     non_optimized_performance = formatter.StatsSpec(
         title=non_optimized_performance_title,

meridian/analysis/review/checks.py

@@ -28,6 +28,7 @@ from meridian.analysis.review import constants as review_constants
 from meridian.analysis.review import results
 from meridian.model import model
 import numpy as np
+import pandas as pd
 
 ConfigType = TypeVar("ConfigType", bound=configs.BaseConfig)
 ResultType = TypeVar("ResultType", bound=results.CheckResult)
@@ -207,6 +208,51 @@ class BayesianPPPCheck(
 # ==============================================================================
 # Check: Goodness of Fit
 # ==============================================================================
+def _set_details_from_gof_dataframe(
+    details: dict[str, float],
+    gof_df: pd.DataFrame,
+    geo_granularity: str,
+    suffix: str | None = None,
+) -> None:
+  """Sets the `details` variable of the GoodnessOfFitCheckResult.
+
+  This method takes a DataFrame containing goodness of fit metrics and pivots it
+  to a Series, which is then added to the `details` variable of the
+  `GoodnessOfFitCheckResult`.
+
+  Args:
+    details: A dictionary to store the goodness of fit metrics in.
+    gof_df: A DataFrame containing predictive accuracy of the whole data (if
+      holdout set is not used) of filtered to a single evaluation set ("all",
+      "train", or "test").
+    geo_granularity: The geo granularity of the data ("geo" or "national").
+    suffix: A suffix to add to the metric names (e.g., "all", "train", "test").
+      If None, the metrics are added without a suffix.
+  """
+  gof_metrics_pivoted = gof_df.pivot(
+      index=constants.GEO_GRANULARITY,
+      columns=constants.METRIC,
+      values=constants.VALUE,
+  )
+  gof_metrics_series = gof_metrics_pivoted.loc[geo_granularity]
+  if suffix is not None:
+    details[f"{review_constants.R_SQUARED}_{suffix}"] = gof_metrics_series[
+        constants.R_SQUARED
+    ]
+    details[f"{review_constants.MAPE}_{suffix}"] = gof_metrics_series[
+        constants.MAPE
+    ]
+    details[f"{review_constants.WMAPE}_{suffix}"] = gof_metrics_series[
+        constants.WMAPE
+    ]
+  else:
+    details[review_constants.R_SQUARED] = gof_metrics_series[
+        constants.R_SQUARED
+    ]
+    details[review_constants.MAPE] = gof_metrics_series[constants.MAPE]
+    details[review_constants.WMAPE] = gof_metrics_series[constants.WMAPE]
+
+
 class GoodnessOfFitCheck(
     BaseCheck[configs.GoodnessOfFitConfig, results.GoodnessOfFitCheckResult]
 ):
@@ -221,38 +267,43 @@ class GoodnessOfFitCheck(
     )
 
     gof_metrics = gof_df[gof_df[constants.GEO_GRANULARITY] == geo_granularity]
-    if constants.EVALUATION_SET_VAR in gof_df.columns:
-      gof_metrics = gof_metrics[
-          gof_metrics[constants.EVALUATION_SET_VAR] == constants.ALL_DATA
-      ]
-
-    gof_metrics_pivoted = gof_metrics.pivot(
-        index=constants.GEO_GRANULARITY,
-        columns=constants.METRIC,
-        values=constants.VALUE,
-    )
-    gof_metrics_series = gof_metrics_pivoted.loc[geo_granularity]
-
-    r_squared = gof_metrics_series[constants.R_SQUARED]
-    mape = gof_metrics_series[constants.MAPE]
-    wmape = gof_metrics_series[constants.WMAPE]
-
-    details = {
-        review_constants.R_SQUARED: r_squared,
-        review_constants.MAPE: mape,
-        review_constants.WMAPE: wmape,
-    }
-
-    if r_squared > 0:
-      return results.GoodnessOfFitCheckResult(
-          case=results.GoodnessOfFitCases.PASS,
-          details=details,
-      )
-    else:  # r_squared <= 0
-      return results.GoodnessOfFitCheckResult(
-          case=results.GoodnessOfFitCases.REVIEW,
+    is_holdout = constants.EVALUATION_SET_VAR in gof_df.columns
+
+    details = {}
+    case = results.GoodnessOfFitCases.PASS
+
+    if is_holdout:
+      for evaluation_set, suffix in [
+          (constants.ALL_DATA, review_constants.ALL_SUFFIX),
+          (constants.TRAIN, review_constants.TRAIN_SUFFIX),
+          (constants.TEST, review_constants.TEST_SUFFIX),
+      ]:
+        set_metrics = gof_metrics[
+            gof_metrics[constants.EVALUATION_SET_VAR] == evaluation_set
+        ]
+        _set_details_from_gof_dataframe(
+            details=details,
+            gof_df=set_metrics,
+            geo_granularity=geo_granularity,
+            suffix=suffix,
+        )
+        if details[f"{review_constants.R_SQUARED}_{suffix}"] <= 0:
+          case = results.GoodnessOfFitCases.REVIEW
+    else:
+      _set_details_from_gof_dataframe(
           details=details,
+          gof_df=gof_metrics,
+          geo_granularity=geo_granularity,
+          suffix=None,
       )
+      if details[review_constants.R_SQUARED] <= 0:
+        case = results.GoodnessOfFitCases.REVIEW
+
+    return results.GoodnessOfFitCheckResult(
+        case=case,
+        details=details,
+        is_holdout=is_holdout,
+    )
 
 
 # ==============================================================================

meridian/analysis/review/constants.py

@@ -32,6 +32,10 @@ NEGATIVE_BASELINE_PROB_REVIEW_THRESHOLD = (
 R_SQUARED = "r_squared"
 MAPE = "mape"
 WMAPE = "wmape"
+ALL_SUFFIX = "all"
+TRAIN_SUFFIX = "train"
+TEST_SUFFIX = "test"
+EVALUATION_SET_SUFFIXES = (ALL_SUFFIX, TRAIN_SUFFIX, TEST_SUFFIX)
 MEAN = "mean"
 VARIANCE = "variance"
 MEDIAN = "median"

meridian/analysis/review/results.py

@@ -319,18 +319,12 @@ class GoodnessOfFitCases(ModelCheckCase, enum.Enum):
 
   PASS = (
       Status.PASS,
-      (
-          "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE ="
-          " {wmape:.4f}."
-      ),
+      "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE = {wmape:.4f}",
       _GOODNESS_OF_FIT_PASS_RECOMMENDATION,
   )
   REVIEW = (
       Status.REVIEW,
-      (
-          "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE ="
-          " {wmape:.4f}."
-      ),
+      "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE = {wmape:.4f}",
       _GOODNESS_OF_FIT_REVIEW_RECOMMENDATION,
   )
 
@@ -348,9 +342,28 @@ class GoodnessOfFitCheckResult(CheckResult):
   """The immutable result of the Goodness of Fit Check."""
 
   case: GoodnessOfFitCases
+  is_holdout: bool = False
 
   def __post_init__(self):
-    if any(
+    if self.is_holdout:
+      required_keys = []
+      for suffix in [
+          constants.ALL_SUFFIX,
+          constants.TRAIN_SUFFIX,
+          constants.TEST_SUFFIX,
+      ]:
+        required_keys.extend([
+            f"{constants.R_SQUARED}_{suffix}",
+            f"{constants.MAPE}_{suffix}",
+            f"{constants.WMAPE}_{suffix}",
+        ])
+      if any(key not in self.details for key in required_keys):
+        raise ValueError(
+            "The message template is missing required formatting arguments for"
+            f" holdout case. Required keys: {required_keys}. Details:"
+            f" {self.details}."
+        )
+    elif any(
         key not in self.details
         for key in (
             constants.R_SQUARED,
@@ -364,6 +377,24 @@ class GoodnessOfFitCheckResult(CheckResult):
           f" {self.details}."
       )
 
+  @property
+  def recommendation(self) -> str:
+    """Returns the check result message."""
+    if self.is_holdout:
+      report_str = (
+          "R-squared = {r_squared_all:.4f} (All),"
+          " {r_squared_train:.4f} (Train), {r_squared_test:.4f} (Test); MAPE"
+          " = {mape_all:.4f} (All), {mape_train:.4f} (Train),"
+          " {mape_test:.4f} (Test); wMAPE = {wmape_all:.4f} (All),"
+          " {wmape_train:.4f} (Train), {wmape_test:.4f} (Test)".format(
+              **self.details
+          )
+      )
+    else:
+      report_str = self.case.message_template.format(**self.details)
+
+    return f"{report_str}. {self.case.recommendation}"
+
 
 # ==============================================================================
 # Check: ROI Consistency

meridian/analysis/summarizer.py

@@ -21,11 +21,11 @@ import os
 import jinja2
 from meridian import constants as c
 from meridian.analysis import analyzer
-from meridian.analysis import formatter
 from meridian.analysis import summary_text
 from meridian.analysis import visualizer
 from meridian.data import time_coordinates as tc
 from meridian.model import model
+from meridian.templates import formatter
 import pandas as pd
 import xarray as xr
 

meridian/analysis/visualizer.py

@@ -22,9 +22,9 @@ import altair as alt
 from meridian import backend
 from meridian import constants as c
 from meridian.analysis import analyzer
-from meridian.analysis import formatter
 from meridian.analysis import summary_text
 from meridian.model import model
+from meridian.templates import formatter
 import numpy as np
 import pandas as pd
 import xarray as xr

meridian/backend/__init__.py

@@ -609,7 +609,37 @@ def _tf_get_seed_data(seed: Any) -> Optional[np.ndarray]:
 
 
 def _jax_convert_to_tensor(data, dtype=None):
-  """Converts data to a JAX array, handling strings as NumPy arrays."""
+  """Converts data to a JAX array, handling strings as NumPy arrays.
+
+  This function explicitly unwraps objects with a `.values` attribute (e.g.,
+  pandas.DataFrame, xarray.DataArray) to access the underlying NumPy array,
+  provided that `.values` is not a method. This takes precedence over the
+  `__array__` protocol.
+
+  It also handles precision mismatches: if `data` is float64 and `dtype` is
+  not specified, and JAX x64 mode is disabled (default), it issues a warning
+  and explicitly casts to float32 to match the backend default and prevent
+  silent precision loss or type errors in downstream operations.
+
+  Args:
+    data: The data to convert.
+    dtype: The desired data type.
+
+  Returns:
+    A JAX array, or a NumPy array if the dtype is a string type.
+  """
+  # Unwrap xarray.DataArray, pandas.Series, and pandas.DataFrame objects.
+  # These objects wrap the underlying NumPy array in a .values attribute.
+  if hasattr(data, "values") and not callable(data.values):
+    data = data.values
+
+  # Convert to numpy array upfront to simplify dtype inspection below.
+  # A standard Python float is 64-bit, and this conversion allows the
+  # subsequent logic to correctly detect and handle potential float64
+  # downcasting for scalar inputs.
+  if isinstance(data, (list, tuple, float)):
+    data = np.array(data)
+
   # JAX does not natively support string tensors in the same way TF does.
   # If a string dtype is requested, or if the data is inherently strings,
   # we fall back to a standard NumPy array.
@@ -623,13 +653,31 @@ def _jax_convert_to_tensor(data, dtype=None):
       # let jax.asarray handle it.
       pass
 
-  is_string_data = isinstance(data, (list, np.ndarray)) and np.array(
-      data
-  ).dtype.kind in ("S", "U")
+  is_string_data = isinstance(data, np.ndarray) and data.dtype.kind in (
+      "S",
+      "U",
+  )
 
-  if is_string_target or (dtype is None and is_string_data):
+  if is_string_target:
     return np.array(data, dtype=dtype)
 
+  if dtype is None and is_string_data:
+    return data
+
+  # If the user provides float64 data but does not request a specific dtype,
+  # and JAX 64-bit mode is disabled (default), JAX would implicitly truncate.
+  # We cast to float32 and warn the user to prevent silent mismatches.
+  if dtype is None:
+    is_float64_input = hasattr(data, "dtype") and data.dtype == np.float64
+    if is_float64_input:
+      if not jax.config.jax_enable_x64:
+        warnings.warn(
+            "Input data is float64. Casting to float32 to match backend "
+            "default precision.",
+            UserWarning,
+        )
+        dtype = jax_ops.float32
+
   return jax_ops.asarray(data, dtype=dtype)
 
 

meridian/backend/test_utils.py

@@ -14,6 +14,7 @@
 
 """Common testing utilities for Meridian, designed to be backend-agnostic."""
 
+import dataclasses
 from typing import Any, Optional
 
 from absl.testing import parameterized
@@ -26,6 +27,8 @@ import numpy as np
 from tensorflow.python.util.protobuf import compare
 # pylint: disable=g-direct-tensorflow-import
 from tensorflow.core.framework import tensor_pb2
+
+
 # pylint: enable=g-direct-tensorflow-import
 
 FieldDescriptor = descriptor.FieldDescriptor
@@ -80,6 +83,75 @@ def assert_allequal(a: ArrayLike, b: ArrayLike, err_msg: str = ""):
   np.testing.assert_array_equal(np.array(a), np.array(b), err_msg=err_msg)
 
 
+def assert_deep_equals(
+    test_case,
+    obj1: Any,
+    obj2: Any,
+    msg: str = "",
+    rtol: float = 1e-5,
+    atol: float = 1e-5,
+):
+  """Recursive equality check handling Dataclasses, Lists, and Backend Tensors.
+
+  Args:
+    test_case: The unittest.TestCase instance (self) to use for assertions.
+    obj1: The first object to compare.
+    obj2: The second object to compare.
+    msg: Optional error message prefix.
+    rtol: Relative tolerance for float comparison.
+    atol: Absolute tolerance for float comparison.
+  """
+  if obj1 is None or obj2 is None:
+    test_case.assertEqual(obj1, obj2, msg=msg)
+    return
+
+  if (
+      hasattr(obj1, "__array__")
+      or hasattr(obj1, "numpy")
+      or isinstance(obj1, (np.ndarray, backend.Tensor))
+  ):
+    arr1 = np.array(obj1)
+    arr2 = np.array(obj2)
+
+    # Check for non-numeric types where atol/rtol don't apply
+    if arr1.dtype.kind in ("U", "S", "O", "b"):
+      np.testing.assert_array_equal(arr1, arr2, err_msg=msg)
+    else:
+      np.testing.assert_allclose(arr1, arr2, err_msg=msg, rtol=rtol, atol=atol)
+    return
+
+  if dataclasses.is_dataclass(obj1):
+    test_case.assertIs(
+        type(obj1),
+        type(obj2),
+        msg=f"{msg} Type mismatch: {type(obj1)} vs {type(obj2)}",
+    )
+    for field in dataclasses.fields(obj1):
+      val1 = getattr(obj1, field.name)
+      val2 = getattr(obj2, field.name)
+      assert_deep_equals(
+          test_case,
+          val1,
+          val2,
+          msg=f"{msg}.{field.name}",
+          rtol=rtol,
+          atol=atol,
+      )
+    return
+
+  if isinstance(obj1, (list, tuple)):
+    test_case.assertIsInstance(obj2, (list, tuple), msg=f"{msg} Type mismatch")
+    test_case.assertEqual(len(obj1), len(obj2), msg=f"{msg} Length mismatch")
+    for i, (item1, item2) in enumerate(zip(obj1, obj2)):
+      assert_deep_equals(
+          test_case, item1, item2, msg=f"{msg}[{i}]", rtol=rtol, atol=atol
+      )
+    return
+
+  # Fallback to standard equality for primitives (int, str, float, etc.)
+  test_case.assertEqual(obj1, obj2, msg=msg)
+
+
 def assert_seed_allequal(a: Any, b: Any, err_msg: str = ""):
   """Backend-agnostic assertion to check if two seed objects are equal."""
   data_a = backend.get_seed_data(a)

meridian/constants.py

@@ -662,6 +662,7 @@ CURRENT_SPEND = 'current_spend'
 
 # Media summary metrics.
 SPEND = 'spend'
+COST = 'cost'
 IMPRESSIONS = 'impressions'
 ROI = 'roi'
 OPTIMIZED_ROI = 'optimized_roi'

meridian/data/load.py

@@ -35,6 +35,8 @@ __all__ = [
     'InputDataLoader',
     'XrDatasetDataLoader',
     'DataFrameDataLoader',
+    'CoordToColumns',
+    'CsvDataLoader',
 ]
 
 

meridian/model/eda/__init__.py

@@ -17,4 +17,3 @@
 from meridian.model.eda import eda_engine
 from meridian.model.eda import eda_outcome
 from meridian.model.eda import eda_spec
-from meridian.model.eda import meridian_eda

meridian/model/eda/constants.py

@@ -14,8 +14,18 @@
 
 """Constants specific to MeridianEDA."""
 
-# EDA Plotting properties
+# EDA Engine constants
+COST_PER_MEDIA_UNIT = 'cost_per_media_unit'
+
 VARIABLE_1 = 'var1'
 VARIABLE_2 = 'var2'
-VARIABLE = 'var'
 CORRELATION = 'correlation'
+ABS_CORRELATION_COL_NAME = 'abs_correlation'
+
+# EDA Plotting properties
+VARIABLE = 'var'
+VALUE = 'value'
+NATIONALIZE = 'nationalize'
+MEDIA_IMPRESSIONS_SCALED = 'media_impressions_scaled'
+IMPRESSION_SHARE_SCALED = 'impression_share_scaled'
+SPEND_SHARE = 'spend_share'