Browse Source

prepare for the branch

define_task
mohamad maheri 2 years ago
parent
commit
34018e3f96
1 changed files with 1 additions and 0 deletions
  1. 1
    0
      learnToLearnTest.py

+ 1
- 0
learnToLearnTest.py View File

@@ -66,6 +66,7 @@ def test(embedding,head, total_dataset, batch_size, num_epoch,adaptation_step=co
predictions = predictions.view(-1)
y_true = query_ys.cpu().detach().numpy()
y_pred = predictions.cpu().detach().numpy()

# ndcgs1.append(float(mz.metrics.NormalizedDiscountedCumulativeGain(k=1)(y_true, y_pred)))
# ndcgs3.append(float(mz.metrics.NormalizedDiscountedCumulativeGain(k=3)(y_true, y_pred)))


Loading…
Cancel
Save