Commit cd4d532c authored by 张彦钊's avatar 张彦钊

add print for test

parent 6e48c18d
......@@ -32,6 +32,10 @@ def test_con_sql(device_id):
megacity_queue = list(map(lambda x: "diary|" + str(x), megacity_queue))
db.close()
print("成功获取日记队列")
print(native_queue)
print(nearby_queue)
print(nation_queue)
print(megacity_queue)
return native_queue, nearby_queue, nation_queue, megacity_queue
else:
print("该用户对应的日记队列为空")
......@@ -82,7 +86,7 @@ def transform_ffm_format(df,queue_name):
ffm_format_pandas = pickle.load(f)
print("1")
data = ffm_format_pandas.native_transform(df)
print("2")
predict_file_name = DIRECTORY_PATH + "result/{0}_{1}.csv".format(device_id, queue_name)
data.to_csv(predict_file_name, index=False, header=None)
print("done ffm")
......
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