Roman Zykov rzykov
rzykov
/ gist:d5aa7e065203a2a7221ce1bbf663f511
Created
August 15, 2017 20:07
— forked from agibsonccc/gist:241034a329bd1e12fa70
Scala nd4j examples
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| package org.deeplearning4j.learn | |
| import java.util.Arrays | |
| import org.nd4j.api.linalg.DSL._ | |
| import org.nd4j.linalg.api.ndarray.INDArray | |
| import org.nd4j.linalg.factory.Nd4j | |
| import org.nd4j.linalg.indexing.{NDArrayIndex, BooleanIndexing} | |
| import org.nd4j.linalg.indexing.conditions.Conditions | |
| import org.nd4j.linalg.indexing.functions.Value |
rzykov
/ Nd4j_lstm.scala
Created
August 21, 2017 08:51
— forked from harperjiang/Nd4j_lstm.scala
Performance comparison of numpy vs nd4j on LSTM implementation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import org.nd4j.linalg.api.ndarray.INDArray | |
| import org.nd4j.linalg.api.ops.impl.broadcast.BroadcastAddOp | |
| import org.nd4j.linalg.api.rng.distribution.impl.UniformDistribution | |
| import org.nd4j.linalg.factory.Nd4j | |
| import scala.util.Random | |
| object Xavier { | |
| def init(shape: Array[Int]): INDArray = { | |
| var n = shape.dropRight(1).product |