23 lines
565 B
Python
Executable file
23 lines
565 B
Python
Executable file
#!/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()
|