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