Commit 6e48c18d authored by 张彦钊's avatar 张彦钊

add print for test

parent 002b5c84
......@@ -80,7 +80,9 @@ def feature_en(x_list, device_id):
def transform_ffm_format(df,queue_name):
with open(DIRECTORY_PATH + "ffm.pkl", "rb") as f:
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