Skip to content

Arfua/tabnet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Data Science Session: TabNet

tabnet-architecture

This session shows how to:

  • Train TabNet Classifier for a multi-task prediction
  • Pre-train TabNet
  • Visualize TabNet training metrics
  • Interpret TabNet feature importances
  • Perform data preparation for TabNet training (categorical feature encoding, filling missing values, etc.)
  • Comparison of TabNet results with XGBoost results
  • Check the similarity of feature distributions between train and test datasets

Pre-requirements

  • python 3.8
  • pip

Requirements

pip install -r requirements.txt

Speaker: Raid Arfua