InterpretableClustering / InterpretableClustering

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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
  • Reddit

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

About


Languages

Language:Jupyter Notebook 76.4%Language:Python 23.6%