(data)
| 150 | |
| 151 | |
| 152 | def get_degree_weights(data): |
| 153 | edge_index_ = to_undirected(data.edge_index) |
| 154 | deg = degree(edge_index_[1]) |
| 155 | deg_col = deg[data.edge_index[1]].to(torch.float32) |
| 156 | scaled_deg_col = torch.log(deg_col) |
| 157 | |
| 158 | return normalize(scaled_deg_col), deg |
| 159 | |
| 160 | |
| 161 | def get_pagerank_weights(data, aggr: str = 'sink', k: int = 10): |
nothing calls this directly
no test coverage detected