Commit daaaffce authored by 赵威's avatar 赵威

update data

parent 4f576c0d
......@@ -10,8 +10,8 @@ from .utils import common_elements, nth_element
def read_csv_data(dataset_path):
device_df = pd.read_csv(dataset_path.joinpath("device.csv"), sep="|")
diary_df = pd.read_csv(dataset_path.joinpath("diary.csv"), sep="|")
click_df = pd.read_csv(dataset_path.joinpath("click.csv"), sep="|")
conversion_df = pd.read_csv(dataset_path.joinpath("click_cvr.csv"), sep="|")
click_df = pd.read_csv(dataset_path.joinpath("diary_click.csv"), sep="|")
conversion_df = pd.read_csv(dataset_path.joinpath("diary_click_cvr.csv"), sep="|")
# TODO remove sample
# return device_df.sample(10000), diary_df.sample(5000), click_df, conversion_df
return device_df, diary_df, click_df, conversion_df
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment