mmpretrain/tools
Ma Zerun b39885d953
[Feature] Support K-fold cross-validation (#563)
* Support to use `indices` to specify which samples to evaluate.

* Add KFoldDataset wrapper

* Rename 'K' to 'num_splits' accroding to sklearn

* Add `kfold-cross-valid.py`

* Add unit tests

* Add help doc and docstring
2022-01-19 18:32:55 +08:00
..
analysis_tools [Refactor] Change `load_json_logs` to `load_json_log`. (#602) 2021-12-14 18:03:27 +08:00
convert_models [Enhance] Add metafile, readme and converted models for MLP-Mixer (#539) 2021-11-24 19:04:19 +08:00
deployment [Feature] Add conversion script from pytroch to mlmodel (#597) 2022-01-13 12:32:52 +08:00
misc [Refactor] Remove useless `import_modules_from_string`. (#544) 2021-11-22 14:01:26 +08:00
visualizations [Feature] Add CAM visualization tool (#577) 2021-12-23 18:53:40 +08:00
dist_test.sh init commit 2020-05-21 21:21:43 +08:00
dist_train.sh init commit 2020-05-21 21:21:43 +08:00
kfold-cross-valid.py [Feature] Support K-fold cross-validation (#563) 2022-01-19 18:32:55 +08:00
slurm_test.sh init commit 2020-05-21 21:21:43 +08:00
slurm_train.sh init commit 2020-05-21 21:21:43 +08:00
test.py [Fix] Fix format error in `test.py` when metric returns `np.ndarray` (#441) 2021-09-29 11:51:19 +08:00
train.py [Refactor] Save class names in best checkpoint created by evaluation hook (#641) 2022-01-13 12:20:59 +08:00