Row-normalize sparse matrix
(mx)
| 54 | |
| 55 | |
| 56 | def normalize(mx): |
| 57 | """Row-normalize sparse matrix""" |
| 58 | rowsum = np.array(mx.sum(1)) |
| 59 | r_inv = np.power(rowsum, -1).flatten() |
| 60 | r_inv[np.isinf(r_inv)] = 0. |
| 61 | r_mat_inv = sp.diags(r_inv) |
| 62 | mx = r_mat_inv.dot(mx) |
| 63 | return mx |
| 64 | |
| 65 | |
| 66 | def accuracy(output, labels): |