public class GloveChange
extends java.lang.Object
implements java.io.Serializable
Constructor and Description |
---|
GloveChange(VocabWord w1,
VocabWord w2,
org.nd4j.linalg.api.ndarray.INDArray w1Update,
org.nd4j.linalg.api.ndarray.INDArray w2Update,
double w1BiasUpdate,
double w2BiasUpdate,
double error,
org.nd4j.linalg.api.ndarray.INDArray w1History,
org.nd4j.linalg.api.ndarray.INDArray w2History,
double w1BiasHistory,
double w2BiasHistory) |
Modifier and Type | Method and Description |
---|---|
void |
apply(GloveWeightLookupTable table)
Apply the changes to the table
|
double |
getError() |
VocabWord |
getW1() |
double |
getW1BiasHistory() |
double |
getW1BiasUpdate() |
org.nd4j.linalg.api.ndarray.INDArray |
getW1History() |
org.nd4j.linalg.api.ndarray.INDArray |
getW1Update() |
VocabWord |
getW2() |
double |
getW2BiasHistory() |
double |
getW2BiasUpdate() |
org.nd4j.linalg.api.ndarray.INDArray |
getW2History() |
org.nd4j.linalg.api.ndarray.INDArray |
getW2Update() |
void |
setError(double error) |
void |
setW1(VocabWord w1) |
void |
setW1BiasHistory(double w1BiasHistory) |
void |
setW1BiasUpdate(double w1BiasUpdate) |
void |
setW1History(org.nd4j.linalg.api.ndarray.INDArray w1History) |
void |
setW1Update(org.nd4j.linalg.api.ndarray.INDArray w1Update) |
void |
setW2(VocabWord w2) |
void |
setW2BiasHistory(double w2BiasHistory) |
void |
setW2BiasUpdate(double w2BiasUpdate) |
void |
setW2History(org.nd4j.linalg.api.ndarray.INDArray w2History) |
void |
setW2Update(org.nd4j.linalg.api.ndarray.INDArray w2Update) |
java.lang.String |
toString() |
public GloveChange(VocabWord w1, VocabWord w2, org.nd4j.linalg.api.ndarray.INDArray w1Update, org.nd4j.linalg.api.ndarray.INDArray w2Update, double w1BiasUpdate, double w2BiasUpdate, double error, org.nd4j.linalg.api.ndarray.INDArray w1History, org.nd4j.linalg.api.ndarray.INDArray w2History, double w1BiasHistory, double w2BiasHistory)
public void apply(GloveWeightLookupTable table)
table
- public org.nd4j.linalg.api.ndarray.INDArray getW1History()
public void setW1History(org.nd4j.linalg.api.ndarray.INDArray w1History)
public org.nd4j.linalg.api.ndarray.INDArray getW2History()
public void setW2History(org.nd4j.linalg.api.ndarray.INDArray w2History)
public double getW1BiasHistory()
public void setW1BiasHistory(double w1BiasHistory)
public double getW2BiasHistory()
public void setW2BiasHistory(double w2BiasHistory)
public VocabWord getW1()
public void setW1(VocabWord w1)
public VocabWord getW2()
public void setW2(VocabWord w2)
public org.nd4j.linalg.api.ndarray.INDArray getW1Update()
public void setW1Update(org.nd4j.linalg.api.ndarray.INDArray w1Update)
public org.nd4j.linalg.api.ndarray.INDArray getW2Update()
public void setW2Update(org.nd4j.linalg.api.ndarray.INDArray w2Update)
public double getW1BiasUpdate()
public void setW1BiasUpdate(double w1BiasUpdate)
public double getW2BiasUpdate()
public void setW2BiasUpdate(double w2BiasUpdate)
public double getError()
public void setError(double error)
public java.lang.String toString()
toString
in class java.lang.Object