Machine Learning Model TrainingML Frameworks

GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

Open Source

About

GraphMAE is designed for generative self-supervised learning on graph data, achieving competitive performance in tasks such as node classification, graph classification, and molecular property prediction. It is implemented in Python and utilizes PyTorch, making it suitable for researchers and practitioners in deep learning and graph neural networks. The method supports various datasets and provides scripts for quick execution of different classification tasks.

Open Source Health

Not enough history
Stars
592
Forks
85
License
Not stated
Last commit
3 years ago
Python

Related Categories