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