Interpretable Clustering
This repository contains the code for Interpretable Clustering on Dynamic Graphs with Recurrent Graph Neural Networks, AAAI 2021.
Data
5 datasets were used in the paper:
- DBLP-E
- DBLP-3
- DBLP-5
- Brain
Requirements
- Python 3,
- PyTorch 1.2.0 or higher,
- TensorFlow==1.11.0,
- dgl-cu101,
- dynamicgem,
- networkx,
- sklearn,
- scipy,
- keras==2.2.4,
- numpy>=1.15.3
Main file
- main.ipynb