Commit 0996dcf2 authored by 赵威's avatar 赵威

prediction

parent c80097d1
......@@ -73,7 +73,7 @@ def main():
# model_predict(test_300, predict_fn)
print("==============================")
device_id = "861601036552944"
# device_id = "861601036552944"
# diary_ids = [
# "16195283", "16838351", "17161073", "17297878", "17307484", "17396235", "16418737", "16995481", "17312201", "12237988"
# ]
......@@ -82,31 +82,32 @@ def main():
df2 = get_diary_df_from_redis()
redis_device_df = device_feature_engineering(df)
redis_diary_df = diary_feature_engineering(df2, from_redis=True)
device_ids = list(redis_device_df["device_id"].values)[:20]
diary_ids = list(redis_diary_df["card_id"].values)
@time_cost
def test1():
user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
model_predict(user1, predict_fn)
@time_cost
def test2():
user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
model_predict(user1, predict_fn)
@time_cost
def test3():
user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
model_predict(user1, predict_fn)
@time_cost
def test4():
user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
model_predict(user1, predict_fn)
@time_cost
def test5():
user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
model_predict(user1, predict_fn)
test1()
......
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