Skip to content

Evaluation of Data to predict if a user would subscribe to a term deposit using NN, evaluated by SHAP

Notifications You must be signed in to change notification settings

Thehacker-arch/SHAPmodel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

10 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Bank Marketing Classification with PyTorch & SHAP (Black Box Model)

A black box model which uses SHAP to evaluate the feature which contributed most to the predict whether a client will subscribe to a term deposit.

πŸ“Œ

  • PyTorch Neural Network
  • SHAP for Feature Importance
  • Data Preprocessing

πŸ”§ Installation

pip install torch pandas shap scikit-learn

About

Evaluation of Data to predict if a user would subscribe to a term deposit using NN, evaluated by SHAP

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published