Splitting into Categorical features

[harshraj22]

DeepLearning/implementations/TabTransformer/data_loader/datasets.py

Line 21 in 60b90a6

CATEGORICAL_COLUMNS = ['gender', 'SeniorCitizen', 'Partner', 'Dependents', 'PhoneService', 'MultipleLines', 'InternetService', 'OnlineSecurity', 'OnlineBackup', 'DeviceProtection', 'TechSupport', 'StreamingTV', 'StreamingMovies', 'Contract', 'PaperlessBilling', 'PaymentMethod']

Since the dataset is specific, it seems ok to use variables to store categorical and continuous variables. For generic case, you may want to use fastai’s cont_cat_split