| import json, random, collections, os |
| random.seed(1234) |
| MANI="/scratch/prep/manifests"; OUT="/root/devhard"; os.makedirs(OUT,exist_ok=True) |
| def load(p): return [json.loads(l) for l in open(p,encoding="utf-8")] |
| dev_all=[] |
| for L in ["lin","sna"]: |
| rows=load(f"{MANI}/waxal_{L}_train.jsonl") |
| by=collections.defaultdict(list) |
| for r in rows: by[r.get("speaker")].append(r) |
| cand=[s for s,v in by.items() if 8<=len(v)<=70] |
| random.shuffle(cand) |
| target=min(400,int(0.05*len(rows))) |
| hold=set(); n=0 |
| for s in cand: |
| if n>=target: break |
| hold.add(s); n+=len(by[s]) |
| dev=[r for r in rows if r.get("speaker") in hold] |
| tmin=[r for r in rows if r.get("speaker") not in hold] |
| for r in dev: r["lang"]=L |
| dev_all+=dev |
| with open(f"{OUT}/devhard_{L}.jsonl","w",encoding="utf-8") as f: |
| for r in dev: f.write(json.dumps(r,ensure_ascii=False)+"\n") |
| with open(f"{OUT}/train_{L}_min.jsonl","w",encoding="utf-8") as f: |
| for r in tmin: f.write(json.dumps(r,ensure_ascii=False)+"\n") |
| print(f"{L}: holdout_spk={len(hold)} devhard={len(dev)} train_min={len(tmin)} (was {len(rows)})") |
| with open(f"{OUT}/devhard_linsna.jsonl","w",encoding="utf-8") as f: |
| for r in dev_all: f.write(json.dumps(r,ensure_ascii=False)+"\n") |
| print("DEVHARD_LINSNA_DONE total",len(dev_all)) |
|
|
