org.deeplearning4j.graph.iterator.parallel

Class WeightedRandomWalkGraphIteratorProvider<V>

java.lang.Object

org.deeplearning4j.graph.iterator.parallel.WeightedRandomWalkGraphIteratorProvider<V>

Type Parameters:

V - Vertex type

All Implemented Interfaces:

GraphWalkIteratorProvider<V>

public class WeightedRandomWalkGraphIteratorProvider<V>
extends java.lang.Object
implements GraphWalkIteratorProvider<V>

Weighted random walk graph iterator provider: given a weighted graph (of type IGraph<?,? extends Number>), split up the generation of weighted random walks for parallel learning. Specifically: with N threads and V vertices: - First iterator generates weighted random walks starting at vertices 0 to V/N - Second iterator generates weighted random walks starting at vertices V/N+1 to 2*V/N - and so on

See Also:

WeightedRandomWalkIterator

Constructor Summary

Constructors

Constructor and Description
WeightedRandomWalkGraphIteratorProvider(IGraph<V,? extends java.lang.Number> graph, int walkLength)
WeightedRandomWalkGraphIteratorProvider(IGraph<V,? extends java.lang.Number> graph, int walkLength, long seed, NoEdgeHandling mode)

Method Summary

Modifier and Type	Method and Description
java.util.List<GraphWalkIterator<V>>	getGraphWalkIterators(int numIterators) Get a list of GraphWalkIterators.

Methods inherited from class java.lang.Object

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

Constructor Detail

WeightedRandomWalkGraphIteratorProvider

public WeightedRandomWalkGraphIteratorProvider(IGraph<V,? extends java.lang.Number> graph,
                                               int walkLength)

WeightedRandomWalkGraphIteratorProvider

public WeightedRandomWalkGraphIteratorProvider(IGraph<V,? extends java.lang.Number> graph,
                                               int walkLength,
                                               long seed,
                                               NoEdgeHandling mode)

Method Detail

getGraphWalkIterators

public java.util.List<GraphWalkIterator<V>> getGraphWalkIterators(int numIterators)

Description copied from interface: GraphWalkIteratorProvider

Get a list of GraphWalkIterators. In general: may return less than the specified number of iterators, (for example, for small networks) but never more than it

Specified by:

getGraphWalkIterators in interface GraphWalkIteratorProvider<V>

Parameters:

numIterators - Number of iterators to return