import pandas as pd
from ChannelAttributionPro import reward_model
token = "yourtoken"
Data = pd.read_csv("https://app.channelattribution.io/data/data_mmm.csv", sep=",")
D_variables = pd.DataFrame({
"variable": [
"direct_searches",
"facebook_impressions",
"facebook_cost",
"google_impressions",
"google_clicks",
"google_cost",
],
"channel": [
"direct",
"facebook_ads",
"facebook_ads",
"google_ads",
"google_ads",
"google_ads",
],
"measure": [
"direct_searches",
"impressions",
"cost",
"impressions",
"clicks",
"cost",
],
})
target = "revenue"
res = reward_model(
Data=Data,
D_variables=D_variables,
target=target,
max_p=12,
seed=1234567,
verbose=1,
password=token,
)
print(res)