GitHub - lucidrains/tab-transformer-pytorch ...
https://github.com/lucidrains/tab-transformer-pytorch31.08.2021 · import torch import torch. nn as nn from tab_transformer_pytorch import TabTransformer cont_mean_std = torch. randn (10, 2) model = TabTransformer ( categories = (10, 5, 6, 5, 8), # tuple containing the number of unique values within each category num_continuous = 10, # number of continuous values dim = 32, # dimension, paper set at 32 dim_out = 1, # binary …