I want to do a graph classification task. Those graphs are directed, and their edges have features. I knew little about graph representation methods, but I did some research, and find most of the works for graph classification seem to be based on undirected graphs. So is there a way to encode this kind of graph? Or any suggestions to encode this kind of graph?


1 Answer 1


Most methods generalize naturally to undirected graphs, some of the simple examples are GCN [1], GIN [2] and GAT [3]. There are more sophisticated methods, but depending on the task, often they don't perform drastically different [4].

It really depends on you specific task what works best for you. You might want to try some different architectures and I'd start with the GCN or GIN. You could opt for a nice library that allows benchmarking all the different architectures quite quickly like deep graph library (DGL) or Torch-Geometric.

  • 1
    $\begingroup$ Thanks very much. This answer gives me clear guidance about what to do next. $\endgroup$
    – han
    Dec 15, 2022 at 15:31
  • $\begingroup$ Nice, I'm glad I could help :) $\endgroup$
    – Chillston
    Dec 15, 2022 at 15:45

You must log in to answer this question.

Not the answer you're looking for? Browse other questions tagged .