Coder Social home page Coder Social logo

graph2gauss-pytorch's Introduction

graph2gauss in PyTorch

This is a pytorch implementation of Deep Gaussian Embedding of Graphs: Unsupervised Inductive Learning via Ranking. Run python g2g.py -h to learn about training options. Below you can see an example run on the citeseer dataset as provided in the implementation by the original authors.

$ python g2g.py --seed 0 --samples 3 --epochs 120 --workers 5 -k 1 citeseer.npz
LR F1 score 0.4491554535256343
Epoch 10 - Loss 35145104.000
Epoch 20 - Loss 28642094.000
Epoch 30 - Loss 21908854.000
Epoch 40 - Loss 17939046.000
Epoch 50 - Loss 14932645.000
LR F1 score 0.7872897465883691
Epoch 60 - Loss 12318501.000
Epoch 70 - Loss 11219567.000
Epoch 80 - Loss 9815305.000
Epoch 90 - Loss 8618428.000
Epoch 100 - Loss 7848496.500
LR F1 score 0.8252173651998809

A significant difference between this and the reference implementation is the Monte-Carlo approximation to the loss function. The authors used node-based sampling whereas I used edge sampling as described on my website.

graph2gauss-pytorch's People

Contributors

martenlienen avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.