Split examples

This commit is contained in:
Andras Schmelczer 2022-05-25 22:54:21 +02:00
parent b4492da2e6
commit df1c9418ee
No known key found for this signature in database
GPG key ID: 39260B5B0614A13E
11 changed files with 401 additions and 252 deletions

23
examples/complex/main_batch.py Executable file
View file

@ -0,0 +1,23 @@
#!/usr/bin/env python3
import json
from random import shuffle
from devtools import debug
from predict_domain import predict_domain
from great_ai import process_batch
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()