classification/config.json

{
  "model_type": "tabfm",
  "version": "1.0.0",
  "task": "classification",
  "framework": "pytorch",
  "embed_dim": 256,
  "max_classes": 10,
  "col_num_blocks": 3,
  "col_nhead": 4,
  "col_num_inds": 256,
  "row_num_blocks": 3,
  "row_nhead": 8,
  "row_num_cls": 8,
  "icl_num_blocks": 24,
  "icl_nhead": 8,
  "ff_factor": 4,
  "feature_group_size": 3
}