R/data.R

In Robyn: Semi-Automated Marketing Mix Modeling (MMM) from Meta Marketing Science

# Copyright (c) Meta Platforms, Inc. and its affiliates.

# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

####################################################################
#' Robyn Dataset: MMM Demo Data
#'
#' Simulated MMM data. Input time series should be daily, weekly or monthly.
#'
#' @family Dataset
#' @docType data
#' @usage data(dt_simulated_weekly)
#' @return data.frame
#' @format An object of class \code{"data.frame"}
#' \describe{
#'   \item{DATE}{Date}
#'   \item{revenue}{Daily total revenue}
#'   \item{tv_S}{Television}
#'   \item{ooh_S}{Out of home}
#'   \item{...}{...}
#' }
#' @examples
#' data(dt_simulated_weekly)
#' head(dt_simulated_weekly)
#' @return Dataframe. Contains simulated dummy dataset to test and run demo.
"dt_simulated_weekly"

# dt_input <- read.csv('data/de_simulated_data.csv')
# save(dt_input, file = "data/dt_input.RData", version = 2)
# dt_simulated_weekly <- as_tibble(dt_simulated_weekly)
# save(dt_simulated_weekly, file = "data/dt_simulated_weekly.RData", version = 2)

####################################################################
#' Robyn Dataset: Holidays by Country
#'
#' Contains \code{prophet}'s "new" default holidays by country.
#' When using own holidays, please keep the header
#' \code{c("ds", "holiday", "country", "year")}.
#'
#'
#' @family Dataset
#' @docType data
#' @usage data(dt_prophet_holidays)
#' @return data.frame
#' @format An object of class \code{"data.frame"}
#' \describe{
#'   \item{ds}{Date}
#'   \item{holiday}{Name of celebrated holiday}
#'   \item{country}{Code for the country (Alpha-2)}
#'   \item{year}{Year of \code{ds}}
#' }
#' @examples
#' data(dt_prophet_holidays)
#' head(dt_prophet_holidays)
#' @return Dataframe. Contains \code{prophet}'s default holidays by country.
"dt_prophet_holidays"

# dt_prophet_holidays <- read.csv("~/Desktop/generated_holidays.csv")
# dt_prophet_holidays <- as_tibble(dt_prophet_holidays)
# lares::missingness(dt_prophet_holidays)
# dt_prophet_holidays <- dplyr::filter(dt_prophet_holidays, !is.na(country))
# save(dt_prophet_holidays, file = "data/dt_prophet_holidays.RData", version = 2)