org.deeplearning4j.graph.models

Interface GraphVectors<V,E>

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

DeepWalk, GraphVectorsImpl

public interface GraphVectors<V,E>
extends java.io.Serializable

Vectors for nodes in a graph. Provides lookup table and convenience methods for graph vectors

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)
double	similarity(Vertex<V> vertex1, Vertex<V> vertex2)
int[]	verticesNearest(int vertexIdx, int top)

Method Detail

getGraph

IGraph<V,E> getGraph()

numVertices

int numVertices()

getVectorSize

int getVectorSize()

getVertexVector

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

getVertexVector

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

verticesNearest

int[] verticesNearest(int vertexIdx,
                      int top)

similarity

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

similarity

double similarity(int vertexIdx1,
                  int vertexIdx2)