treat_model function

Specify the propensity score model for a CDE treatment