weifantt / AutoFS

AutoFS

Source code for the paper, "AutoFS: Automated feature selection via diversity-aware interactive reinforcement learning", in ICDM 2020.

Overview

AutoFS is a reinforcement learning based feature selection method.

Citation

If you find our work interesting, you can the paper as

@inproceedings{fan2020autofs,
  title={Autofs: Automated feature selection via diversity-aware interactive reinforcement learning},
  author={Fan, Wei and Liu, Kunpeng and Liu, Hao and Wang, Pengyang and Ge, Yong and Fu, Yanjie},
  booktitle={2020 IEEE International Conference on Data Mining (ICDM)},
  pages={1008--1013},
  year={2020},
  organization={IEEE}
}