Understanding the neural network

[Malaikah-Javed]

1. Could you tell me how the code handles the 6 probabilities? If i have to draw the Neural Network, how would i do that? e.g how many input features, no. of hidden layers (and their names and the flow of the NN), number of neurons in output layer.

2. I am trying to get the weights of input 3D image after pooling. But i am unable to understand why the the code runs 5 times and prints different pool_output["hidden"] values those 5 times. This is why i need the drawing of the Neural Network.

(I know the weights are not updated when running inference. This is why i want to extract the weights during testing inside of the code in the SybilNet class.)
If you know how i can extract those weights, it will be very helpful! Thank you!