03INFER.ipynb,能否公开一下 about kdd-multimodalities-recall HOT 1 CLOSED

      def test_step(self, batch, batch_nb):
          tokens, attn1, rois, attn2, qid, pid = batch
          y_hat = self.predict(tokens, attn1, rois, attn2)
          return {'y_hat': y_hat, 'qid':qid, 'pid':pid}
      def test_epoch_end(self, outputs):
          predict=[]
          for o in outputs:
              qid = o['qid'].cpu().numpy()[0]
              pid = o['pid'].cpu().numpy()[0]
              scores = o['y_hat'].cpu().numpy()[0]
              predict.append((qid,pid,scores))
          self.write_prediction('predict',predict,'predictions.pt')
          return {'test_loss': torch.tensor(0), 'log': {'ndcg5':0}}
      def test_dataloader(self):
          return DataLoader(test, batch_size=1)