Rename

examples/simple/main_batch.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python3
+
+import json
+from random import shuffle
+
+from devtools import debug
+from great_ai import process_batch
+
+from predict_domain import predict_domain
+
+if __name__ == "__main__":
+    with open(".cache/data-1/s2-corpus-323.json") as f:
+        raw = json.load(f)
+
+    shuffle(raw)
+    data = {f'{r["title"]} {r["abstract"]}': r["domain"] for r in raw[:10]}
+
+    results = process_batch(predict_domain, data.keys())
+
+    for predicted, actual in zip(results, data.values()):
+        print(", ".join(actual))
+        debug(predicted)
+        print()