org.deeplearning4j.graph.models.embeddings

Class GraphVectorsImpl<V,E>

java.lang.Object

org.deeplearning4j.graph.models.embeddings.GraphVectorsImpl<V,E>

All Implemented Interfaces:

java.io.Serializable, GraphVectors<V,E>

Direct Known Subclasses:

DeepWalk

public class GraphVectorsImpl<V,E>
extends java.lang.Object
implements GraphVectors<V,E>

Base implementation for GraphVectors. Used in DeepWalk, and also when loading graph vectors from file.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected IGraph<V,E>	graph
protected GraphVectorLookupTable	lookupTable

Constructor Summary

Constructors

Constructor and Description
GraphVectorsImpl()

Method Summary

Modifier and Type	Method and Description
IGraph<V,E>	getGraph()
int	getVectorSize()
org.nd4j.linalg.api.ndarray.INDArray	getVertexVector(int vertexIdx)
org.nd4j.linalg.api.ndarray.INDArray	getVertexVector(Vertex<V> vertex)
int	numVertices()
double	similarity(int vertexIdx1, int vertexIdx2) Returns the cosine similarity of the vector representations of two vertices in the graph, given the indices of these verticies
double	similarity(Vertex<V> vertex1, Vertex<V> vertex2) Returns the cosine similarity of the vector representations of two vertices in the graph
int[]	verticesNearest(int vertexIdx, int top)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

graph

protected IGraph<V,E> graph

lookupTable

protected GraphVectorLookupTable lookupTable

Constructor Detail

GraphVectorsImpl

public GraphVectorsImpl()

Method Detail

getGraph

public IGraph<V,E> getGraph()

Specified by:

getGraph in interface GraphVectors<V,E>

numVertices

public int numVertices()

Specified by:

numVertices in interface GraphVectors<V,E>

getVectorSize

public int getVectorSize()

Specified by:

getVectorSize in interface GraphVectors<V,E>

getVertexVector

public org.nd4j.linalg.api.ndarray.INDArray getVertexVector(Vertex<V> vertex)

Specified by:

getVertexVector in interface GraphVectors<V,E>

getVertexVector

public org.nd4j.linalg.api.ndarray.INDArray getVertexVector(int vertexIdx)

Specified by:

getVertexVector in interface GraphVectors<V,E>

verticesNearest

public int[] verticesNearest(int vertexIdx,
                             int top)

Specified by:

verticesNearest in interface GraphVectors<V,E>

similarity

public double similarity(Vertex<V> vertex1,
                         Vertex<V> vertex2)

Returns the cosine similarity of the vector representations of two vertices in the graph

Specified by:

similarity in interface GraphVectors<V,E>

Returns:

Cosine similarity of two vertices

similarity

public double similarity(int vertexIdx1,
                         int vertexIdx2)

Returns the cosine similarity of the vector representations of two vertices in the graph, given the indices of these verticies

Specified by:

similarity in interface GraphVectors<V,E>

Returns:

Cosine similarity of two vertices