|
--- |
|
license: openrail |
|
tags: |
|
- graphs |
|
pipeline_tag: graph-ml |
|
--- |
|
# Model Card for RGCN |
|
|
|
RGCN is a GNN that can be used to learn node embeddings for Graphs. These embeddings can be used for downstream learning tasks. |
|
|
|
## Model Description |
|
|
|
RGCN is a GNN made up of RGCNConv from PyTorch Geometric |
|
|
|
|
|
## Model Sources |
|
|
|
<!-- Provide the basic links for the model. --> |
|
|
|
- **Example:** [Github](https://github.com/pyg-team/pytorch_geometric/blob/master/examples/rgcn.py) |
|
- **Paper:** [Paper](https://arxiv.org/abs/1703.06103) |
|
- **Documentation:** [Link](https://pytorch-geometric.readthedocs.io/en/latest/generated/torch_geometric.nn.conv.RGCNConv.html#torch_geometric.nn.conv.RGCNConv) |
|
|
|
|
|
## Direct Use |
|
|
|
This model should be used to learn node embeddings for Graphs. These embeddings can be used for downstream learning tasks. |
|
|
|
|
|
## How to Get Started with the Model |
|
|
|
See example |
|
|
