From 30f5a7901cf4dbaeff72635c3f52ca7cf87f0974 Mon Sep 17 00:00:00 2001 From: Marcin Date: Tue, 14 May 2019 15:34:26 +0200 Subject: [PATCH] Save predictions --- ulmfit/train_clas.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py index 534bd37..33aecc1 100644 --- a/ulmfit/train_clas.py +++ b/ulmfit/train_clas.py @@ -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}")