Scikit-learn
English
wwood commited on
Commit
092c39c
·
1 Parent(s): c207928

Use xgboost by default.

Browse files
Files changed (1) hide show
  1. 17_apply_to_proteome.py +1 -1
17_apply_to_proteome.py CHANGED
@@ -110,7 +110,7 @@ if __name__ == '__main__':
110
  parent_parser.add_argument('--modal-keggs', help='modal keggs', default=os.path.join(data_dir, 'ModalKEGGs.tsv'))
111
  parent_parser.add_argument('--modal-keggs-with-names', help='modal keggs with names', default=os.path.join(data_dir, 'modal_keggs_with_names.csv'))
112
  # ./12_apply_model.py --model the.model -x the.x --training-data-header <(head the.training-data-header)
113
- parent_parser.add_argument('--models', nargs='+', help='models to use')
114
 
115
  parent_parser.add_argument('--training-data-header', help='header of training data', default=os.path.join(data_dir, 'training_data_header.tsv'))
116
 
 
110
  parent_parser.add_argument('--modal-keggs', help='modal keggs', default=os.path.join(data_dir, 'ModalKEGGs.tsv'))
111
  parent_parser.add_argument('--modal-keggs-with-names', help='modal keggs with names', default=os.path.join(data_dir, 'modal_keggs_with_names.csv'))
112
  # ./12_apply_model.py --model the.model -x the.x --training-data-header <(head the.training-data-header)
113
+ parent_parser.add_argument('--models', nargs='+', help='models to use', default=[os.path.join(data_dir, '..', 'XGBoost.model')])
114
 
115
  parent_parser.add_argument('--training-data-header', help='header of training data', default=os.path.join(data_dir, 'training_data_header.tsv'))
116