meridian.data.load.XrDatasetDataLoader

从 xarray.Dataset 对象读取数据。

继承自：InputDataLoader

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

查看源代码

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

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