org.nd4j.parameterserver.distributed.training.chains

Class SkipGramChain

java.lang.Object

org.nd4j.parameterserver.distributed.training.chains.SkipGramChain

All Implemented Interfaces:

Chain

public class SkipGramChain
extends java.lang.Object
implements Chain

Chain implementation for SkipGram

Author:

[email protected]

Field Summary

Fields

Modifier and Type	Field and Description
protected DotAggregation	dotAggregation
protected long	frameId
protected long	originatorId
protected SkipGramRequestMessage	requestMessage
protected long	taskId

Constructor Summary

Constructors

Constructor and Description
SkipGramChain(long originatorId, long taskId, long frameId)
SkipGramChain(long taskId, SkipGramRequestMessage message)
SkipGramChain(SkipGramRequestMessage message)

Method Summary

Modifier and Type	Method and Description
void	addElement(VoidMessage message)
long	getTaskId()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.parameterserver.distributed.messages.Chain

getOriginatorId

Field Detail

originatorId

protected long originatorId

taskId

protected long taskId

frameId

protected long frameId

requestMessage

protected SkipGramRequestMessage requestMessage

dotAggregation

protected DotAggregation dotAggregation

Constructor Detail

SkipGramChain

public SkipGramChain(long originatorId,
                     long taskId,
                     long frameId)

SkipGramChain

public SkipGramChain(@NonNull
                     SkipGramRequestMessage message)

SkipGramChain

public SkipGramChain(long taskId,
                     @NonNull
                     SkipGramRequestMessage message)

Method Detail

getTaskId

public long getTaskId()

Specified by:

getTaskId in interface Chain

addElement

public void addElement(VoidMessage message)

Specified by:

addElement in interface Chain