config.json

{
  "attribute_encoder_type": "onehot",
  "continuous_head": "mse",
  "ignore_attributes": false,
  "mlp_hidden_dim": 256,
  "onehot_attr_input_dim": 336,
  "qoi_dim": 1,
  "seq_hidden_size": 128,
  "seq_input_dim": 103,
  "seq_num_layers": 4,
  "seq_type": "ssm",
  "ssm_pool": [],
  "text_encoder_name": "distilbert-base-uncased",
  "text_finetune_only_specific_layers": true,
  "text_freeze_encoder": false,
  "timestamp_embedding_size": 32
}