加入新推出的 Discord 社区，展开实时讨论，获得同行支持，并直接与 Meridian 团队互动！

meridian.data.load.XrDatasetDataLoader

从 xarray.Dataset 对象读取数据。

meridian.data.load.XrDatasetDataLoader(
    dataset: xr.Dataset,
    kpi_type: str,
    name_mapping: (Mapping[str, str] | None) = None
)

  data_loader = XrDatasetDataLoader(pickle.loads('data.pickle'))
  data = data_loader.load()

实参
`dataset`	一个 `xarray.Dataset` 对象，包含输入数据。
`kpi_type`	一个字符串，用于表示相应 KPI 是 `'revenue'` 类型还是 `'non-revenue'` 类型。如果 `kpi_type` 为 `'non-revenue'`，且存在 `revenue_per_kpi`，系统会使用投资回报率校准，并针对收入运行分析。如果同一 `kpi_type` 中不存在 `revenue_per_kpi`，系统会使用自定义投资回报率校准，并针对 KPI 运行分析。
`name_mapping`	一个可选的字典，其键为 `input` 数据集中的当前坐标或数组名称，其值为所需的坐标（`geo`、`time`、`media_time`、`media_channel` 和/或 `rf_channel`、`control_variable`、`non_media_channel`、`organic_media_channel`、`organic_rf_channel`）或数组名称（`kpi`、`revenue_per_kpi`、`media`、`media_spend` 和/或 `rf_spend`、`controls`、`population`、`non_media_treatments`、`organic_media`、`organic_reach`、`organic_frequency`）。如果 `input` 数据集中的名称与所需名称不同，则必须提供映射，否则会系统抛出错误。

属性
`dataset`	一个 `xr.Dataset` 对象，包含输入数据。
`kpi_type`	一个字符串，用于表示相应 KPI 是 `'revenue'` 类型还是 `'non-revenue'` 类型。如果 `kpi_type` 为 `'non-revenue'`，且存在 `revenue_per_kpi`，系统会使用投资回报率校准，并针对收入运行分析。如果同一 `kpi_type` 中不存在 `revenue_per_kpi`，系统会使用自定义投资回报率校准，并针对 KPI 运行分析。

方法

load() -> meridian.data.input_data.InputData

返回一个 InputData 对象，其中包含数据集中的数据。