(X, topks)
| 38 | |
| 39 | |
| 40 | def test_one_batch(X, topks): |
| 41 | sorted_items = X[0].numpy() |
| 42 | groundTrue = X[1] |
| 43 | r = getLabel(groundTrue, sorted_items) |
| 44 | recall, ndcg = [], [] |
| 45 | for k in topks: |
| 46 | recall.append(Recall_ATk(groundTrue, r, k)) |
| 47 | ndcg.append(NDCGatK_r(groundTrue, r, k)) |
| 48 | return {'recall': np.array(recall), |
| 49 | 'ndcg': np.array(ndcg)} |
| 50 | |
| 51 | |
| 52 | def eval_PyTorch(model, data_generator, Ks): |
no test coverage detected