| @@ -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))) | |||