dmlc / GNNLens2

Visualization tool for Graph Neural Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GNNLens2 is an interactive visualization tool for graph neural networks (GNN). It allows seamless integration with deep graph library (DGL) and can meet your various visualization requirements for presentation, analysis and model explanation. It is an open source version of GNNLens with simplification and extension.

A video demo is available here. Switch the video quality for the best viewing experience.

Installation

Requirements

You can install Flask-CORS with

pip install -U flask-cors

Installation for the latest stable version

pip install Flask==2.0.3
pip install gnnlens

Installation from source

If you want to try experimental features, you can install from source as follows:

git clone https://github.com/dmlc/GNNLens2.git
cd GNNLens2/python
python setup.py install

Verifying successful installation

Once you have installed the package, you can verify the success of installation with

import gnnlens

print(gnnlens.__version__)
# 0.1.0

Tutorials

We provide a set of tutorials to get you started with the library:

Team

HKUST VisLab: Zhihua Jin, Huamin Qu

AWS Shanghai AI Lab: Mufei Li, Wanru Zhao (work done during internship), Jian Zhang, Minjie Wang

SMU: Yong Wang

About

Visualization tool for Graph Neural Networks

License:Apache License 2.0


Languages

Language:TypeScript 76.7%Language:Python 16.7%Language:CSS 2.9%Language:HTML 2.1%Language:JavaScript 1.6%