Java 访问deeplearning4j中所有层的权重和原始激活

Java 访问deeplearning4j中所有层的权重和原始激活,java,neural-network,deeplearning4j,Java,Neural Network,Deeplearning4j,我的目标是可视化对图像进行分类的模型。对于可视化,我需要每个层的原始激活/输出。在进行预测时,是否有方法获取这些信息?此外,如果有一种获取权重的方法,这将非常有用。但是,这只是可选的 可视化模型是动态构建的,将用于对MNIST和EMNIST数据集的图像进行分类 示范模型的model.summary(): ======================================================================= LayerName (LayerType) n

我的目标是可视化对图像进行分类的模型。对于可视化,我需要每个层的原始激活/输出。在进行预测时,是否有方法获取这些信息?此外,如果有一种获取权重的方法,这将非常有用。但是,这只是可选的

可视化模型是动态构建的,将用于对MNIST和EMNIST数据集的图像进行分类

示范模型的model.summary():

=======================================================================
LayerName (LayerType)   nIn,nOut   TotalParams   ParamsShape           
=======================================================================
layer0 (DenseLayer)     784,200    157.000       W:{784,200}, b:{1,200}
layer1 (DenseLayer)     200,100    20.100        W:{200,100}, b:{1,100}
layer2 (OutputLayer)    100,10     1.010         W:{100,10}, b:{1,10}  
-----------------------------------------------------------------------
            Total Parameters:  178.110
        Trainable Parameters:  178.110
           Frozen Parameters:  0
=======================================================================
图像分类代码:

INDArray reshaped = reshapeImage(image);
int predictedIndex = model.predict(reshaped)[0];
double conf = model.output(reshaped).getDouble(predictedIndex);
如果您需要更多信息/代码片段,请告诉我