def test_full_connection_layer(self):
     nn = GraphLabNeuralNetBuilder()
     nn.add_flatten_layer()
     nn.add_full_connection_layer(10)
     nn.add_full_connection_layer(100)
 def test_flatten_layer(self):
     nn = GraphLabNeuralNetBuilder()
     nn.add_flatten_layer()