dgllge 1.0.1

Graph Embedding Models

Graph Embedding

Introduction

This module provides the services and implementation for various graph embedding models.

Getting Started

Installation

You can install the DGLL Graph Embedding version 1.0.1 from PyPI as:

pip install dgllge

Usage and Tutorial

input graph

# import module
import ge


# Set Path to Data
data_dir = "Your Path to Data"
dataset = "File/Dataset Name"


# Load a Graph
inputGraph = ge.loadGraph(data_dir, dataset)

Configurable Parameter for Graph Embedding

embedDim = 2 # embedding size
numbOfWalksPerVertex = 2 # walks per vertex
walkLength = 4 # walk lenght
lr =0.025 # learning rate
windowSize = 3 # window size

Choose One of the Following Graph Embedding Models

# DeepWalk
rw = ge.DeepWalk(inputGraph, walkLength=walkLength, embedDim=embedDim, numbOfWalksPerVertex=numbOfWalksPerVertex, \
              windowSize=windowSize, lr = lr)
              

# Node2Vec
rw = ge.Node2vec(inputGraph, walkLength=walkLength, embedDim=embedDim, numbOfWalksPerVertex=numbOfWalksPerVertex, \
               windowSize=windowSize, lr=lr, p = 0.5, q = 0.8)

# Struc2Vec
rw = ge.Struc2Vec(inputGraph, walkLength=walkLength, embedDim=embedDim, numbOfWalksPerVertex=numbOfWalksPerVertex, \
              windowSize=windowSize, lr = lr)

Skip Gram model

modelSkipGram = ge.SkipGramModel(rw.totalNodes, rw.embedDim)

Want Node Embedding or Edge Embedding

# Learning Node Embedding
model = rw.learnNodeEmbedding(modelSkipGram)

# Learning Edge Embedding
model = rw.learnEdgeEmbedding(modelSkipGram)

Plot Embedding

ge.plot_2DEmbedding(rw)

Save Embedding to Disk

ge.saveEmbedding(data_dir, dataset, rw)

Generate Embedding for a Specific Node or Edge

node1 = 35
node2 = 40

# Get Embedding for a node
emb = rw.getNodeEmbedding(node1)
print("Node Embedding", emb)

# Get Embedding for an edge
emb = rw.getEdgeEmbedding(node1, node2)
print("Edge Embedding", emb)

License

Distributed under the MIT License. See LICENSE for more information.

(back to top)

Acknowledgments

• Data and Knowlege Engineering Lab (DKE)

(back to top)

Links

Project's GitHub Link: @Graph-Embedding

Project's PyPI Link: @dgllge

(back to top)