提交 6d9d2e1e 编写于 作者: W wangjiawei04

pre-commit code test=serving

上级 0cbffcc7
FILE(GLOB common_srcs ${CMAKE_CURRENT_LIST_DIR}/constant.cpp)
LIST(APPEND pdserving_srcs ${common_srcs})
......@@ -28,11 +28,8 @@ batch = 1
buf_size = 100
dataset = criteo.CriteoDataset()
dataset.setup(1000001)
test_filelists = [
"{}/part-test".format(sys.argv[2])
]
reader = dataset.infer_reader(test_filelists, batch,
buf_size)
test_filelists = ["{}/part-test".format(sys.argv[2])]
reader = dataset.infer_reader(test_filelists, batch, buf_size)
label_list = []
prob_list = []
start = time.time()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册