Basic reference implementation of DeepMinCut in Pytorch.
pytorch >= 1.0
Execute train.sh
to run the example on Cora dataset.
- Additional datasets can be found from OpenNE https://github.com/thunlp/OpenNE
- Word adjacency matrices can be constructed using https://github.com/ziyin-dl/word-embedding-dimensionality-selection