Skip to content

Commit 6ce5c2e

Browse files
committed
credit
1 parent 5314485 commit 6ce5c2e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
This package consists of a small extension library of highly optimized graph cluster algorithms for the use in [PyTorch](http://pytorch.org/).
1717
The package consists of the following clustering algorithms:
1818

19-
* **[Graclus](#graclus)**
19+
* **[Graclus](#graclus)** from Dhillon *et al.*: [Weighted Graph Cuts without Eigenvectors: A Multilevel Approach](http://www.cs.utexas.edu/users/inderjit/public_papers/multilevel_pami.pdf) (PAMI 2007)
2020
* **[VoxelGrid](#voxelgrid)**
2121

2222
All included operations work on varying data types and are implemented both for CPU and GPU.
@@ -34,6 +34,7 @@ pip install cffi torch-cluster
3434
## Graclus
3535

3636
A greedy clustering algorithm of picking an unmarked vertex and matching it with one its unmarked neighbors (that maximizes its edge weight).
37+
The GPU algorithm is adapted from Fagginger Auer and Bisseling: [A GPU Algorithm for Greedy Graph Matching](http://www.staff.science.uu.nl/~bisse101/Articles/match12.pdf) (LNCS 2012)
3738

3839
```python
3940
import torch

0 commit comments

Comments
 (0)