Merge remote-tracking branch 'fork/master'
commit
6b557f2441
|
@ -108,7 +108,7 @@ public class Tsne {
|
||||||
|
|
||||||
int n = X.rows();
|
int n = X.rows();
|
||||||
// FIXME: this is wrong, another distribution required here
|
// FIXME: this is wrong, another distribution required here
|
||||||
Y = randn(X.rows(), targetDimensions, Nd4j.getRandom());
|
Y = Nd4j.randn(X.dataType(), X.rows(), targetDimensions);
|
||||||
INDArray dY = Nd4j.zeros(n, targetDimensions);
|
INDArray dY = Nd4j.zeros(n, targetDimensions);
|
||||||
INDArray iY = Nd4j.zeros(n, targetDimensions);
|
INDArray iY = Nd4j.zeros(n, targetDimensions);
|
||||||
INDArray gains = Nd4j.ones(n, targetDimensions);
|
INDArray gains = Nd4j.ones(n, targetDimensions);
|
||||||
|
|
Loading…
Reference in New Issue