Skip to content

anpolol/Benchmarking-Dataset

Repository files navigation

Benchmarking-Loss-Functions

About the project

We tested a number of local structure preserving loss functions and their influence on node classification task.

Installation

  1. Run pip install -r requirements.txt
  2. Install Torch 1.11.0
  3. Install Torch Geometric in current experiments PyG==2.1.0 is used.

How to use

examples are in NodeClassificaton.jpynb

Available loss functions

In some loss functions we added negative sampling optimization (VERSE, LINE, DeepWalk, Node2Vec)

Available convolutions

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published