Build fix - tsne random method signature (#259)
Signed-off-by: AlexDBlack <blacka101@gmail.com>master
parent
bc2a7dd7ae
commit
c1bcbdda18
|
@ -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