This repository contains implementation of Deep Feature Selection algorithm on PyTorch from this article
Data was taken from here
In the .ipynb file also have done comparision Neural Network approach with RandomForestClassifier from scikit-learn.
Simple Neural Network approach consist of 3 layers: DFS and two Linear:
Trained model is saved and accessible for use