Paschalis Melissas

Results 11 comments of Paschalis Melissas

``` def predict_prob_embed(self, X, Y, eval=True): loader_te = DataLoader(self.handler(X, Y, transform=self.args['test_transform']), shuffle=False, **self.args['loader_te_args']) probs = torch.zeros([len(Y), self.clf.n_label]) embeddings = torch.zeros([len(Y), self.clf.get_embedding_dim()]) if eval: self.clf.eval() with torch.no_grad(): for x, y, idxs...