Applying embeddings on categorical data

I am working on a dataset which comprises of both image and tabular data. So I wanted to train them both together in a single model. For images I am using pretrained model from torchvision and for tabular data fastai’s TabularModel() but I was a little confused about the arguments taken in by TabularModel. Can someone explain it by an example

See here for an example:

While this is in v2, the idea in v1 (And parameters) is still the same. Calculate the size of your embeddings, then you’ll need your number of continuous variables and your number of class output (1 for regression).

1 Like