diff --git a/Workflows/basic.py b/Workflows/basic.py index 6d439b86e5fb4db50f5c107735d8daacc6f221b6..c9e6a0daddc3b73dc2e03d72a9772497b284c8c0 100644 --- a/Workflows/basic.py +++ b/Workflows/basic.py @@ -96,7 +96,7 @@ def train_valid_workflow(seg, ts_lst_pkl, d, m_file): obc.gen_k_folds(5, class_field=d['ref_db']['fields'][-1], augment=d['training']['augment_if_missing'], - min_samples_per_class=d['training']['augment_if_missing']) + min_samples_per_class=d['training']['min_samples_per_class']) if 'export_training_base' in d['training'].keys() and d['training']['export_training_base'] is True: obc.save_training_base('{}/_side/training_base.pkl'.format(os.path.join(d['output_path'], d['chain_name'])))