Skip to content

Commit

Permalink
more pep8
Browse files Browse the repository at this point in the history
  • Loading branch information
Eh2406 committed Apr 9, 2018
1 parent 5a3d390 commit d10e194
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion urbansim/models/dcm.py
Original file line number Diff line number Diff line change
Expand Up @@ -433,7 +433,8 @@ def fit(self, choosers, alternatives, current_choice):
'the input columns.')

self.log_likelihoods, self.fit_parameters = mnl.mnl_estimate(
model_design.as_matrix(), chosen, self.sample_size, self.normalize, l1=self.l1, l2=self.l2)
model_design.as_matrix(), chosen, self.sample_size,
self.normalize, l1=self.l1, l2=self.l2)
self.fit_parameters.index = model_design.columns

logger.debug('finish: fit LCM model {}'.format(self.name))
Expand Down
3 changes: 2 additions & 1 deletion urbansim/urbanchoice/mnl.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,8 @@ def mnl_loglik(beta, data, chosen, numalts, weights=None, lcgrad=False,
return -1 * loglik, -1 * gradarr


def mnl_simulate(data, coeff, numalts, normalization_mean=0.0, normalization_std=1.0, GPU=False, returnprobs=True):
def mnl_simulate(data, coeff, numalts, normalization_mean=0.0, normalization_std=1.0, GPU=False,
returnprobs=True):
"""
Get the probabilities for each chooser choosing between `numalts`
alternatives.
Expand Down

0 comments on commit d10e194

Please sign in to comment.