def build_model(X, p_drop): l1 = nnbuilder.denseLayer(X, 256, activation=nn.rectify, w_init=nn.XavierNormal(root_two)) d1 = nnbuilder.dropoutLayer(l1, p_drop) l2 = nnbuilder.denseLayer(d1, 256, activation=nn.rectify, w_init=nn.XavierNormal(root_two)) d2 = nnbuilder.dropoutLayer(l2, p_drop) out = nnbuilder.denseLayer(d2, 10, activation=nn.softmax) return out