org.deeplearning4j.graph.models.embeddings

Interface GraphVectorLookupTable

All Known Implementing Classes:

InMemoryGraphLookupTable

public interface GraphVectorLookupTable

Lookup table for vector representations of the vertices in a graph

Method Summary

Modifier and Type	Method and Description
int	getNumVertices() Returns the number of vertices in the graph
org.nd4j.linalg.api.ndarray.INDArray	getVector(int idx) Get the vector for the vertex with index idx
void	iterate(int first, int second) Conduct learning given a pair of vertices (in and out)
void	resetWeights() Reset (randomize) the weights.
void	setLearningRate(double learningRate) Set the learning rate
int	vectorSize() The size of the vector representations

Method Detail

vectorSize

int vectorSize()

The size of the vector representations

resetWeights

void resetWeights()

Reset (randomize) the weights.

iterate

void iterate(int first,
             int second)

Conduct learning given a pair of vertices (in and out)

getVector

org.nd4j.linalg.api.ndarray.INDArray getVector(int idx)

Get the vector for the vertex with index idx

setLearningRate

void setLearningRate(double learningRate)

Set the learning rate

getNumVertices

int getNumVertices()

Returns the number of vertices in the graph