Build fix - tsne random method signature (#259)

Signed-off-by: AlexDBlack <blacka101@gmail.com>
master
Alex Black 2019-09-13 13:40:46 +10:00 committed by GitHub
parent bc2a7dd7ae
commit c1bcbdda18
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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);