Tag: deeplearning4j

Deeplearning4j LSTM输出大小

我的情况 – 在输入我有List<List<Float>> （单词表示向量的列表）。 而且 – 从一个序列输出一个Double 。 所以我建立下一个结构（第一个索引 – 例子编号，第二个句子编号，第三个词向量元素编号）： http : //pastebin.com/KGdjwnki 并在输出： http : //pastebin.com/fY8zrxEL 但是，当我将其中一个（ http://pastebin.com/wvFFC4Hw ）转化为model.output时 – 我得到的矢量[0.25, 0.24, 0.25, 0.25] 0.25，0.24，0.25，0.25 [0.25, 0.24, 0.25, 0.25] ，而不是一个值。 什么可能是错的？ 附加的代码（在Kotlin）。 classCount是一个。 import org.deeplearning4j.nn.multilayer.MultiLayerNetwork import org.deeplearning4j.nn.conf.NeuralNetConfiguration.Builder import org.deeplearning4j.nn.api.OptimizationAlgorithm import org.deeplearning4j.nn.conf.Updater import org.deeplearning4j.nn.weights.WeightInit import org.deeplearning4j.nn.conf.layers.GravesLSTM import org.deeplearning4j.nn.conf.layers.RnnOutputLayer import org.deeplearning4j.nn.conf.BackpropType import org.nd4j.linalg.api.ndarray.INDArray import org.nd4j.linalg.cpu.nativecpu.NDArray […]