disease_model/adhd/config.json

{
  "architectures": [
    "KMaxMeanCNN"
  ],
  "auto_map": {
    "AutoConfig": "model_hf_disease.DiseaseConfig",
    "AutoModel": "model_hf_disease.KMaxMeanCNN"
  },
  "dropout": 0.2,
  "filter_num": 50,
  "filter_sizes": [
    2,
    3,
    4,
    5,
    6
  ],
  "in_dim": 38,
  "max_pooling_k": 5,
  "model_type": "kmax_mean_cnn",
  "torch_dtype": "float32",
  "transformers_version": "4.46.2"
}