Skip to content

Instantly share code, notes, and snippets.

View thibaudmartinez's full-sized avatar

Thibaud Martinez thibaudmartinez

View GitHub Profile
@thibaudmartinez
thibaudmartinez / rgcn-node-attributes.py
Last active August 8, 2023 09:53
Relational Graph Convolutional Network for entity classification taking into account node attributes.
import torch
from torch import nn
from torch import optim
from torch.nn import functional as F
from dgl import function as fn
import dgl.init
from dgl import DGLGraph
from dgl.contrib.data import load_data
import numpy as np
import networkx as nx