Skip to content

G
gm_strategy_cvr
Commit 17b5c5c8 authored by 赵威's avatar 赵威
Browse files
Options

update df size

parent 6f03163c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/models/esmm/fe/tractate_fe.py
View file @ 17b5c5c8
......@@ -27,8 +27,7 @@ def read_csv_data(dataset_path):
tractate_df = pd.read_csv(dataset_path.joinpath("tractate.csv"), sep="|")
click_df = pd.read_csv(dataset_path.joinpath("tractate_click.csv"), sep="|")
conversion_df = pd.read_csv(dataset_path.joinpath("tractate_click_cvr.csv"), sep="|")
# TODO
return tractate_df.sample(5000), click_df.sample(10000), conversion_df
return tractate_df, click_df, conversion_df
def get_tractate_from_redis():
......
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