kien-vu/adaptiveFS

Adaptive feature selection using reinforcement learning

A repository for adaptive feature selection using Reinforcement Learning.

Requirements: python 3.7.1 pytorch 1.4

pretrain guesser:

python pretrain_guesser.py

train RL agent:

python ddqn_questionnaire.py