This is a source code for AAAI 2019 paper Classification with Costly Features using Deep Reinforcement Learning wrote by Jaromír Janisch, Tomáš Pevný and Viliam Lisý: paper / slides / poster / code / blog

Updated version available: There is an enhanced version of the article under name Classification with Costly Features as a Sequential Decision-Making Problem (paper), which analyzes more settings (hard budget, lagrangian optimization of lambda and missing features). The code is available in the lagrange branch of this repository.

Cite as:

  title={Classification with Costly Features using Deep Reinforcement Learning},
  author={Janisch, Jaromír and Pevný, Tomáš and Lisý, Viliam},
  booktitle={AAAI Conference on Artificial Intelligence},