ver Feb26th

statistics of action types of exp_som branch
This commit is contained in:
David Chang
2024-02-26 20:50:08 +08:00
parent e72d938f2f
commit 1ff9f5c38b

22
action_type_analysis.py Normal file
View File

@@ -0,0 +1,22 @@
import json
import glob
from typing import Dict, Counter
from typing import Any
import collections
action_counter: Counter[str] = collections.Counter()
for trjctr in glob.glob("snapshots/exp_som/exp_trajectory/**/*.json", recursive=True):
with open(trjctr) as f:
for l in f:
step: Dict[str, Any] = json.loads(l)
#print(step)
for rsp in step.get("action", "").splitlines():
if rsp.startswith("pyautogui."):
action_counter[rsp[10:rsp.find("(")]] += 1
elif rsp in {"WAIT", "FAIL", "DONE"}:
action_counter[rsp] += 1
for k, nb in sorted(action_counter.items(), key=(lambda itm: itm[1]), reverse=True):
print(k, nb)