ProtoPatient / config.json
row56's picture
Update config.json
ea54ef8 verified
{
"text_config": {"pretrained_model_name_or_path": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext"},
"architectures": ["ProtoForMultiLabelClassification"],
"model_type": "proto",
"num_labels": 519,
"reduce_hidden_size": 256,
"use_attention": true,
"use_sigmoid": true,
"label_order_path": "label_order.json",
"prototype_vector_path": "prototype_vectors",
"attention_vector_path": "attention_vectors",
"prototype_vector_path": "model.safetensors",
"attention_vector_path": "model.safetensors"
}