lucidrains / tab-transformer-pytorch

Implementation of TabTransformer, attention network for tabular data, in Pytorch
MIT License
812 stars 103 forks source link

TypeError: can only concatenate str (not "int") to str #21

Open manu-chauhan opened 1 year ago

manu-chauhan commented 1 year ago

Hi,

I am getting this issue when I try to run pred = model(x_categ, x_cont)

One more question, we are supposed to give exact number of unique values for each categorical column in the order they appear in data (categories) and the total number of continuous columns (num_continuous)... is there anything else I am missing out?

model = TabTransformer(categories = tuple(cat_list), num_continuous = len(continuous_cols), dim = 32, dim_out = 61, depth = 6, heads = 8, attn_dropout = 0.1, ff_dropout = 0.1, mlp_hidden_mults = (4, 2), mlp_act = nn.ReLU())

minjun-tilda commented 8 months ago

There is only variable name.. We need to know about exact value.