Skip to content

G
gm_strategy_cvr
Commit c8b30d12 authored by 赵威's avatar 赵威
Browse files
Options

update predict size

parent 47a2e455
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
src/online_prediction.py
View file @ c8b30d12
...@@ -20,6 +20,8 @@ def consume_kafka(device_dict, diary_dict, tractate_dict, diary_predict_fn, trac ...@@ -20,6 +20,8 @@ def consume_kafka(device_dict, diary_dict, tractate_dict, diary_predict_fn, trac
consumer.poll() consumer.poll()
consumer.seek_to_end() consumer.seek_to_end()
predict_size = 300
try: try:
for msg in consumer: for msg in consumer:
data = json.loads(msg.value) data = json.loads(msg.value)
...@@ -28,9 +30,9 @@ def consume_kafka(device_dict, diary_dict, tractate_dict, diary_predict_fn, trac ...@@ -28,9 +30,9 @@ def consume_kafka(device_dict, diary_dict, tractate_dict, diary_predict_fn, trac
event_cn = data.get("event_cn", "") event_cn = data.get("event_cn", "")
# if device_id == "androidid_a25a1129c0b38f7b": # if device_id == "androidid_a25a1129c0b38f7b":
print(device_id + " " + log_time + " " + event_cn + " " + str(datetime.now())) print(device_id + " " + log_time + " " + event_cn + " " + str(datetime.now()))
offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn, size=300) offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn, size=predict_size)
print("---------------------") print("-------------------")
offline_predict_tractate(device_id, device_dict, tractate_dict, tractate_predict_fn, size=300) offline_predict_tractate(device_id, device_dict, tractate_dict, tractate_predict_fn, size=predict_size)
print("=========================================\n") print("=========================================\n")
except Exception as e: except Exception as e:
print(e) print(e)
......
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