model = Model(
    ProbabilisticDense(200, input_dim=X_train.shape[-1]),
    Activation('relu'),
    ProbabilisticDense(200),
    Activation('relu'),