Build fix - tsne random method signature (#259)
Signed-off-by: AlexDBlack <blacka101@gmail.com>
This commit is contained in:
parent
bc2a7dd7ae
commit
c1bcbdda18
@ -108,7 +108,7 @@ public class Tsne {
|
||||
|
||||
int n = X.rows();
|
||||
// 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 iY = Nd4j.zeros(n, targetDimensions);
|
||||
INDArray gains = Nd4j.ones(n, targetDimensions);
|
||||
|
Loading…
x
Reference in New Issue
Block a user