itsmelisssssaa's picture
Add APT classification model and configuration
8ffb7bf
raw
history blame contribute delete
220 Bytes
{
"model_type": "cysecbert-apt",
"architectures": ["CySecBERTMaxPerformance"],
"num_classes": 12,
"hidden_size": 768,
"max_position_embeddings": 512,
"model_class": "APTClassifier",
"framework": "pytorch"
}