Commit 79a750fa authored by 赵威's avatar 赵威

add printer

parent 0996dcf2
......@@ -87,28 +87,63 @@ def main():
@time_cost
def test1():
time_1 = timeit.default_timer()
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
total_1 = (timeit.default_timer() - time_1)
print("join df cost {:.5f}s".format(total_1))
time_1 = timeit.default_timer()
model_predict(user1, predict_fn)
total_1 = (timeit.default_timer() - time_1)
print("total prediction cost {:.5f}s".format(total_1))
@time_cost
def test2():
time_1 = timeit.default_timer()
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
total_1 = (timeit.default_timer() - time_1)
print("join df cost {:.5f}s".format(total_1))
time_1 = timeit.default_timer()
model_predict(user1, predict_fn)
total_1 = (timeit.default_timer() - time_1)
print("total prediction cost {:.5f}s".format(total_1))
@time_cost
def test3():
time_1 = timeit.default_timer()
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
total_1 = (timeit.default_timer() - time_1)
print("join df cost {:.5f}s".format(total_1))
time_1 = timeit.default_timer()
model_predict(user1, predict_fn)
total_1 = (timeit.default_timer() - time_1)
print("total prediction cost {:.5f}s".format(total_1))
@time_cost
def test4():
time_1 = timeit.default_timer()
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
total_1 = (timeit.default_timer() - time_1)
print("join df cost {:.5f}s".format(total_1))
time_1 = timeit.default_timer()
model_predict(user1, predict_fn)
total_1 = (timeit.default_timer() - time_1)
print("total prediction cost {:.5f}s".format(total_1))
@time_cost
def test5():
time_1 = timeit.default_timer()
user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
total_1 = (timeit.default_timer() - time_1)
print("join df cost {:.5f}s".format(total_1))
time_1 = timeit.default_timer()
model_predict(user1, predict_fn)
total_1 = (timeit.default_timer() - time_1)
print("total prediction cost {:.5f}s".format(total_1))
test1()
test2()
......@@ -117,7 +152,7 @@ def main():
test5()
total_time = (time.time() - time_begin) / 60
print("cost {:.2f} mins at {}".format(total_time, datetime.now()))
print("total cost {:.2f} mins at {}".format(total_time, datetime.now()))
if __name__ == "__main__":
......
import timeit
import tensorflow as tf
from tensorflow import feature_column as fc
from tensorflow.python.estimator.canned import head as head_lib
from tensorflow.python.ops.losses import losses
import timeit
def build_deep_layer(net, params):
......@@ -115,4 +116,5 @@ def model_predict(inputs, predict_fn):
total_1 = (timeit.default_timer() - time_1)
print("prediction cost {:.5f}s".format(total_1))
# print(predictions)
# TODO get the diary ids
return predictions
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