andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Add log_argument

Browse source 
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-04-09 22:02:00 +02:00
parent 09137e146c
commit 69d5c4f1f6
27 changed files with 332 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

23
example/main.py
View file

@ -1,23 +0,0 @@
import json
from random import shuffle
from devtools import debug
from predict_domain import predict_domain
from good_ai import process_batch, serve
if __name__ == "__main__":
serve(predict_domain)
with open(".cache/ss-data-0/s2-corpus-1583.json") as f:
raw = json.load(f)
shuffle(raw)
data = {f'{r["title"]} {r["abstract"]}': r["domain"] for r in raw[:5]}
results = process_batch(predict_domain, data.keys())
for predicted, actual in zip(results, data.values()):
print(", ".join(actual))
debug(predicted)
print()