Skip to content

antoniofilipovic/node-classification-methods-analysis

Repository files navigation

node-classification-methods-analysis

Part of master's degree.

Worked on analyisis of node classification techniques in graph machine learning. Tested on various datasets Node2Vec, GraphSAGE, GCN and GAT.

Here are some of the results:

  • GAT accuracy 82.1% on CORA dataset - visualized with tSNE method:

  • GCN accuracy of 76% on CORA dataset - visualized with tSNE method:

  • Node2Vec accuracy of 73% on CORA - visualized with tSNE method:

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages