Save predictions

2026-08-24 12:15:19 +08:00 · 2019-05-14 15:34:40 +02:00
parent 7955db37ad
commit 30f5a7901c
1 changed files with 1 additions and 0 deletions
@@ -197,6 +197,7 @@ class CLSHyperParams(LMHyperParams):
            ds = data_clas.valid_dl
        else:
            raise AttributeError(f"Unrecognized mode {mode}, valid options: test, valid, train optionally dev==valid")
+        np.save(self.model_dir / f"preds-on-{mode}.npy", probs.cpu().numpy())
        results = learn.validate(ds)
        print(f"Model: {self.name}")
        print(f"Validation on: {mode}")