Skip to content

Commit

Permalink
Change model generation defaults to reflect results from feature expl…
Browse files Browse the repository at this point in the history
…oration
  • Loading branch information
Profir-Petru Partachi committed May 3, 2018
1 parent d88fe08 commit bb4ff2d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions backend/generate_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,8 @@ def main(args):
}
features = ['cosine_tc', 'report_size', 'branch_size', 'files_touched_by_pr', 'developer_normalised_lag']
print('Loaded repository %s' % repo.name)
linker = Linker(net_size_in_days=30, min_tok_len=3, undersample_multiplicity=1000, feature_config=config,
predictions_between_updates=1000, stopwords=stopwords)
linker = Linker(net_size_in_days=30, min_tok_len=2, undersample_multiplicity=1e20, feature_config=config,
predictions_between_updates=1e10, stopwords=stopwords)
linker.fit(repo, truth, features=features)
print('Trained Random Forest classifier')
out_path = os.path.join(os.getcwd(), 'models', repo.name[1:].translate({ord(c): '_' for c in '\\/'}))
Expand Down

0 comments on commit bb4ff2d

Please sign in to comment.