Spaces:
Sleeping
Sleeping
File size: 1,017 Bytes
6f4f21f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import pandas as pd
import numpy as np
from data_utils.data_simulation import UpliftSimulationReady
class EDASimulationReady:
def __init__(self, files_path):
self.files_path = files_path
def load_conversions(self, file_name):
uplift_simulation = UpliftSimulationReady(self.files_path)
df = uplift_simulation.load_uplift_data(file_name)
sum_conversions = df.pivot_table(values=['conversion','discounted_price','benefit'],
index='treatment_group_key',
aggfunc=[np.sum],
margins=False)
mean_conversions = df.pivot_table(values=['conversion','discounted_price','benefit'],
index='treatment_group_key',
aggfunc=[np.mean],
margins=False)
# save to csv
sum_conversions.to_csv(self.files_path + 'sum_conversions.csv')
mean_conversions.to_csv(self.files_path + 'mean_conversions.csv')
return sum_conversions, mean_conversions
|