search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
21.51k stars 3.69k forks source link

using edge_weight into shape classification #2346

Closed Rishabh201911010 closed 3 years ago

Rishabh201911010 commented 3 years ago

From now on, we recommend using our discussion forum (https://github.com/rusty1s/pytorch_geometric/discussions) for general questions.

❓ Questions & Help

Dear rusty1s, I am trying to classify shape which are 3D using GCN . For Edge_weight I am using Laplace Beltrami operator. Now I have to incorporate this edge weight into ModelNet10 dataset. How can I do this?

rusty1s commented 3 years ago

You can add edge weights to your dataset via the transform and pre_transform arguments of a dataset. Furthermore, there exists many GNNs that can take edge_weight into account, such as GraphConv and GCNConv.

Rishabh201911010 commented 3 years ago

Thank You Sir.