Skip to content

maksym33/FeatureImportanceDL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FeatureImportanceDL

Contains example of a deep-learning based dual-net feature selection method from https://arxiv.org/abs/2010.08973 :

This is an embedded method for supervised tasks: after training, it is able to give predictions (better then vanilla architecture) while also return population-wise (global) feature importances.

See example.py script that shows how the method works for a basic XOR dataset. Some examples for MNIST dataset (digit 3 & 8 differentiation):

MNIST importance Top features

Superimposed on mean 3 Superimposed on mean 8

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages