Simplify _linear_model_fn

* reduce line count using getattr() on layers 'weighted_sum...' rather than two explicit calls

*  create a variable that has a reference to the correct function

as suggested

* use a better name for the logit function

* use regular if statement

* be more explicit in if statement

* fix whitespace

* whitespace fix