fayssica / AAAI-2019-AFS

The code of the AAAI-19 paper "AFS: An Attention-based mechanism for Supervised Feature Selection".

AAAI-2019-AFS

The code of the AAAI-19 paper "AFS: An Attention-based mechanism for Supervised Feature Selection".

• AFS.py:the main implementation of the proposed approach AFS
• run_AFS: gives an example to show the full proceduce of training and evaluating the proposed approach AFS

The datasets used in the experiment:

• 1 http://yann.lecun.com/exdb/mnist/
• 2 http://csc.lsu.edu/~saikat/n-mnist/
• 3 http://featureselection.asu.edu

Run an example

 python main.py  

@inproceeding{  
  title     = {AFS: An Attention-based mechanism for Supervised Feature Selection},  

  author    = {Ning Gui, Danni Ge, Ziyin Hu},  

  booktitle = {Proceedings of the Thirty-Third {AAAI} Conference on Artificial Intelligence,  
             January 27-February 1, 2019, Hilton Hawaiian Village, Honolulu, Hawaii, {USA.}},  

  year      = {2019}  
}