Understanding of the layers parameter in get_tabular_learner

(Tony Hung) #1

I haven’t been able to find any documentation on the layer parameter in get_tabular_learner function

learn = get_tabular_learner(data, layers=[500,100], metrics=accuracy)

does anyone know what layers is and what it does

(Jan) #2

It refers to the sizes of the hidden fully connected layers between the input (after embedding) and before the classification layer. The number of hidden layers is determined by the length of the list. So in your case you’ll have two hidden fully connected layers of size 500 and 100 respectively.

Edit: This snippet from tabular/models.py should make it clearer …