pytorch_geometric_signed_directed

Documentation | Case Study | Data Set Descriptions | Installation | Data Structures | External Resources | Paper

PyTorch Geometric Signed Directed is a signed and directed extension library for PyTorch Geometric. It follows the package structure in PyTorch Geometric Temporal.

We also provide detailed examples in the examples folder.

Citing

If you find PyTorch Geometric Signed Directed useful in your research, please consider adding the following citation:

@inproceedings{he2024pytorch,
  title={Pytorch Geometric Signed Directed: A software package on graph neural networks for signed and directed graphs},
  author={He, Yixuan and Zhang, Xitong and Huang, Junjie and Rozemberczki, Benedek and Cucuringu, Mihai and Reinert, Gesine},
  booktitle={Learning on Graphs Conference},
  pages={12--1},
  year={2024},
  organization={PMLR}
}

Methods Included

In detail, the following signed or directed graph neural networks, as well as related methods designed for signed or directed netwroks, were implemented.

Directed Unsigned Network Models and Layers

• MagNet_node_classification from Zhang et al.: MagNet: A Neural Network for Directed Graphs. (NeurIPS 2021)

• DiGCL from Tong et al.: Directed Graph Contrastive Learning. (NeurIPS 2021)

• DiGCN_Inception_Block_node_classification from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• DIGRAC_node_clustering from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (LoG 2022)

• DGCN_node_classification from Tong et al.: Directed Graph Convolutional Network. (ArXiv 2020)

• DiGCN_node_classification from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• MagNet_link_prediction from Zhang et al.: MagNet: A Neural Network for Directed Graphs. (NeurIPS 2021)

• DiGCN_link_prediction from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• DiGCN_Inception_Block_link_prediction from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• DGCN_link_prediction from Tong et al.: Directed Graph Convolutional Network. (ArXiv 2020)

• DiGCN_Inception_Block from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• DGCNConv from Tong et al.: Directed Graph Convolutional Network. (ArXiv 2020)

• MagNetConv from Zhang et al.: MagNet: A Neural Network for Directed Graphs. (NeurIPS 2021)

• DiGCNConv from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• DIMPA from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (LoG 2022)

Signed (Directed) Network Models and Layers

• SSSNET_node_clustering from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• SDGNN from Huang et al.: SDGNN: Learning Node Representation for Signed Directed Networks (AAAI 2021)

• SiGAT from Huang et al.: Signed Graph Attention Networks (ICANN 2019)

• MSGNN_link_prediction from He et al.: MSGNN: A Spectral Graph Neural Network Based on a Novel Magnetic Signed Laplacian. (LoG 2022)

• MSGNN_node_classification from He et al.: MSGNN: A Spectral Graph Neural Network Based on a Novel Magnetic Signed Laplacian. (LoG 2022)

• MSConv from He et al.: MSGNN: A Spectral Graph Neural Network Based on a Novel Magnetic Signed Laplacian. (LoG 2022)

• SSSNET_link_prediction adapted from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• SNEA from Li et al.: Learning Signed Network Embedding via Graph Attention (AAAI 2020)

• SGCN from Derr et al.: Signed Graph Convolutional Networks (ICDM 2018)

• SNEAConv from Li et al.: Learning Signed Network Embedding via Graph Attention (AAAI 2020)

• SGCNConv from Derr et al.: Signed Graph Convolutional Network (ICDM 2018)

• SIMPA from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

Network Generation Methods

• Signed Stochastic Block Model(SSBM) from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• Polarized Signed Stochastic Block Model(POL-SSBM) from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• Directed Stochastic Block Model(DSBM) from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (LoG 2022)

• Signed Directed Stochastic Block Model(SDSBM) from He et al.: MSGNN: A Spectral Graph Neural Network Based on a Novel Magnetic Signed Laplacian. (LoG 2022)

Data Loaders and Classes

• load_signed_real_data to load signed (directed) real-world data sets.

• load_directed_real_data to load directed unsigned real-world data sets.

• SignedData Signed Data Class.

• DirectedData Directed Data Class.

• SSSNET_signed_real_data to load signed real-world data sets from the SSSNET paper.

• SDGNN_signed_real_data to load signed real-world data sets from the SDGNN paper.

• MSGNN_signed_directed_real_data to load signed directed real-world data sets from the MSGNN paper.

• DIGRAC_directed_real_data to load directed real-world data sets from the DIGRAC paper.

• Telegram to load the Telegram data set.

• Cora_ml to load the Cora_ML data set.

• Citeseer to load the CiteSeer data set.

• WikiCS to load the WikiCS data set.

• WikipediaNetwork to load the WikipediaNetwork data set.

Task-Specific Objectives and Evaluation Methods

• Probabilistic Balanced Normalized Loss from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• Probabilistic Imbalance Objective from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (LoG 2022)

• Probabilistic Balanced Ratio Loss from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• Unhappy Ratio from He et al.: SSSNET: Semi-Supervised Signed Network Clustering (SDM 2022)

• link_sign_prediction_logistic_function for signed networks' link sign prediction task.

• link_sign_direction_prediction_logistic_function for signed directed networks' link prediction task.

• triplet_loss_node_classification for triplet loss in the node classification task.

• Sign_Triangle_Loss from Huang et al.: SDGNN: Learning Node Representation for Signed Directed Networks (AAAI 2021)

• Sign_Direction_Loss from Huang et al.: SDGNN: Learning Node Representation for Signed Directed Networks (AAAI 2021)

• Sign_Product_Entropy_Loss from Huang et al.: SDGNN: Learning Node Representation for Signed Directed Networks (AAAI 2021)

• Link_Sign_Product_Loss from Huang et al.: Signed Graph Attention Networks (ICANN 2019)

• Link_Sign_Entropy_Loss from Derr et al.: Signed Graph Convolutional Network (ICDM 2018)

• Sign_Structure_Loss

Utilities and Preprocessing Methods

• node_class_split to split nodes into training set etc..

• link_class_split to split edges into training set etc..

• get_magnetic_Laplacian from from Zhang et al.: MagNet: A Neural Network for Directed Graphs. (NeurIPS 2021)

• get_magnetic_signed_Laplacian from He et al.: MSGNN: A Spectral Graph Neural Network Based on a Novel Magnetic Signed Laplacian. (LoG 2022)

• get_appr_directed_adj from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• meta_graph_generation from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (ArXiv 2021)

• extract_network from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (LoG 2022)

• directed_features_in_out from Tong et al.: Directed Graph Convolutional Network. (ArXiv 2020)

• get_second_directed_adj from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• cal_fast_appr from Tong et al.: Digraph Inception Convolutional Networks. (NeurIPS 2020)

• scipy_sparse_to_torch_sparse from He et al.: DIGRAC: Digraph Clustering Based on Flow Imbalance. (LoG 2022)

• create spectral features

Head over to our documentation to find out more! If you notice anything unexpected, please open an issue. If you are missing a specific method, feel free to open a feature request.

Installation

Binaries are provided for Python version >= 3.7 and NetworkX version < 2.7.

After installing PyTorch and PyG, simply run

pip install torch-geometric-signed-directed

Running tests

$ python setup.py test

License

• MIT License