File size: 872 Bytes
1c0cccd
 
da65732
 
 
1c0cccd
da65732
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
---
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