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

add print for test

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