uplift_modeling / data_utils /data_simulation.py
howardroark's picture
initial commit
6f4f21f
raw
history blame contribute delete
702 Bytes
import pandas as pd
class UpliftSimulationReady:
def __init__(self, files_path):
self.files_path = files_path
def load_user_profiles(self, file_name):
user_profiles = pd.read_csv(self.files_path + file_name)
return user_profiles
def load_uplift_data(self, file_name):
uplift_data = pd.read_csv(self.files_path + file_name)
return uplift_data
def load_irrelevant_data(self, file_name):
irrelevant_data = pd.read_csv(self.files_path + file_name)
return irrelevant_data
def load_other_data(self, file_name):
transaction_data = pd.read_csv(self.files_path + file_name)
return transaction_data