Graph data augmentation library
Project description
Mooon: Graph Data Augmentation Library
人有悲欢离合,月有阴晴圆缺。 ———— 苏轼《水调歌头》
Why "Mooon"?
Graph with data augmentations, is like the moon, now dark, now full.
Quick Tour
- Functional API
from mooon import drop_edge
edge_index, edge_weight = drop_edge(edge_index, p=0.5)
edge_index, edge_weight = drop_edge(edge_index, edge_weight, p=0.5)
- Module Layer
from mooon import DropEdge
drop_edge = DropEdge(p=0.5)
edge_index, edge_weight = drop_edge(edge_index)
edge_index, edge_weight = drop_edge(edge_index, edge_weight)
🚀 Installation
Please make sure you have installed PyTorch and PyTorch Geometric (PyG).
# Coming soon
pip install -U mooon
or
# Recommended
git clone https://github.com/EdisonLeeeee/Mooon.git && cd Mooon
pip install -e . --verbose
where -e
means "editable" mode so you don't have to reinstall every time you make changes.
Roadmap
Note: this is an ongoing project, please feel free to contact me for collaboration.
- Based on PyTorch
- Support only PyG
- High-level class and low-level functional API
- Seamlessly integrated into existing code written by PyG
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mooon-0.0.1b0.tar.gz
(7.0 kB
view hashes)