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
