(test_data, pred_data)
| 2 | import numpy as np |
| 3 | |
| 4 | def getLabel(test_data, pred_data): |
| 5 | r = [] |
| 6 | for i in range(len(test_data)): |
| 7 | groundTrue = test_data[i] |
| 8 | predictTopK = pred_data[i] |
| 9 | pred = list(map(lambda x: x in groundTrue, predictTopK)) |
| 10 | pred = np.array(pred).astype("float") |
| 11 | r.append(pred) |
| 12 | return np.array(r).astype('float') |
| 13 | |
| 14 | |
| 15 | def Recall_ATk(test_data, r, k): |