Test for predict, add tf_admm_glm