我正在研究的模型应该同时预测相当多的变量(>1000)。因此,我希望在每个输出的网络末端都有一个小型神经网络。
为了紧凑地做到这一点,我想找到一种方法在 Tensorflow 框架内的神经网络的两层之间创建稀疏的可训练连接。
连接矩阵中只有一小部分应该是可训练的:只有参数是块对角线的一部分。
例如:
连接矩阵如下:
可训练参数应该位于1的位置。
python machine-learning tensorflow
machine-learning ×1
python ×1
tensorflow ×1