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