{ | |
"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 | |
} |