(self, dataset, segment="test")
| 312 | json.dump(info, f) |
| 313 | |
| 314 | def predict(self, dataset, segment="test"): |
| 315 | if not self.fitted: |
| 316 | raise ValueError("model is not fitted yet!") |
| 317 | |
| 318 | test_set = dataset.prepare(segment) |
| 319 | |
| 320 | metrics, preds = self.test_epoch(test_set, return_pred=True) |
| 321 | self.logger.info("test metrics: %s" % metrics) |
| 322 | |
| 323 | return preds |
| 324 | |
| 325 | |
| 326 | class LSTM(nn.Module): |
nothing calls this directly
no test coverage detected