可视化3D分类

时间:2018-10-02 20:58:57

标签: python numpy matplotlib graph classification

我建立了一个仅在一个简单的数据集上进行训练的前馈神经网络。现在,我想在3d图形中将其预测为“ 0”时和将其预测为“ 1”时将其可视化。我想像有一种方法可以着色3d图形的区域。红色表示预测为“ 0”,绿色表示预测为“ 1”。

分别是第0层和第1层以及第1层和第2层之间的权重。

[[ 0.58973111  0.34269001  0.44871886 -1.67827511]
 [-2.42426727 -0.5926487  -3.92009576  2.60742128]
 [ 0.2404434   3.73626231  1.32101205 -0.74959622]]

[[-33.06206479]
 [  6.36657289]
 [-32.49209798]
 [ 10.89765273]]

如何可视化?

这是我的数据集在3d图中的图像,我成像后以某种方式可以将预测1和0的区域着色:

enter image description here

0 个答案:

没有答案