"""Exact ground truth (top-100 by cosine) for each corpus size."""
import numpy as np, time
from common import corpus, queries, ground_truth, NP

q = queries()
for n in (100_000, 250_000, 500_000):
    t = time.time()
    gt = ground_truth(corpus(n), q, k=100)
    np.save(f"{NP}/gt_{n}.npy", gt)
    print(n, "done", round(time.time() - t, 1), "s")