Merge branch 'reproduce-poleval' of https://github.com/n-waves/ulmfit-multilingual into reproduce-poleval

2026-09-09 11:27:26 +08:00 · 2019-05-15 18:00:45 +02:00
parent 7c9129670e 30f5a7901c
commit 75b934bb83
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}")