--- title: Message Passing keywords: fastai sidebar: home_sidebar summary: "Implementation of message passing graph network layers like LightGCN, LR-GCCF etc." description: "Implementation of message passing graph network layers like LightGCN, LR-GCCF etc." nb_path: "nbs/models/layers/models.layers.message_passing.ipynb" ---
import pandas as pd
train = pd.DataFrame(
{'userId':[1,1,2,2,3,4,5],
'itemId':[1,2,1,3,2,4,5],
'rating':[4,5,2,5,3,2,4]}
)
train
from torch_geometric.data import Data
E = nn.Parameter(torch.empty(5, 5))
edge_user = torch.tensor(train[train['rating']>3]['userId'].values-1)
edge_item = torch.tensor(train[train['rating']>3]['itemId'].values-1)
edge_ = torch.stack((torch.cat((edge_user,edge_item),0),torch.cat((edge_item,edge_user),0)),0)
data_p = Data(edge_index=edge_)
torch.random.manual_seed(0)
lightgconv = LightGConv()
lightgconv(E, data_p.edge_index)
torch.random.manual_seed(0)
lrgccf = LRGCCF(5,5)
lrgccf(E, data_p.edge_index)