tabnet icon indicating copy to clipboard operation
tabnet copied to clipboard

hyperparameter tuning questions

Open 3SMMZRjWgS opened this issue 2 years ago • 2 comments

Sorry if my 2 questions below sound noobish:

  1. In the original TabNet paper under "Guidelines for hyperparameters" section, it talks about

"An optimal choice of γ can have a major role on the performance."

Can someone tell me which one of the hyperparameters in tabnet_config deals with this γ?

  1. Is there a package that can help auto tune the hyperparameters for a Tabnet?

3SMMZRjWgS avatar Mar 17 '22 01:03 3SMMZRjWgS

Hello @3SMMZRjWgS

  1. The gamma parameter in the paper and original code has been renamed feature_reusage in tabnet_config() function. It defaults to 1.3

  2. You can tune this hyper-parameter through any method you want using the {workflow} package for example. For his you need to use the tabnet() command to fit tabnet models using {parsnip} machinery. You can derive the workflow test if that helps.

cregouby avatar Mar 17 '22 08:03 cregouby

@cregouby: Thank you for the explanation and showing the way!

3SMMZRjWgS avatar Mar 17 '22 21:03 3SMMZRjWgS