boblaros's picture
Upload folder using huggingface_hub
3b979cc verified
Raw History Blame Contribute Delete
2.49 kB
{
"model_id": "complexity-bilstm-final-v1",
"domain": "complexity",
"display_name": "Complexity BiLSTM Final v1",
"description": "Final BiLSTM with additive attention for CEFR-based text complexity classification.",
"is_active": true,
"priority": 5,
"framework": {
"type": "pytorch",
"task": "sequence-classification",
"library": "torch",
"architecture": "bilstm-attention",
"embeddings": "fasttext-wiki-news-subwords-300"
},
"artifacts": {
"weights": [
"models/glove_bilstm_complexity_train_pool_20260307_163358/best_model/best.pt"
],
"tokenizer": [],
"config": [
"outputs/metrics/cfg_20260310_2125.json",
"models/glove_bilstm_complexity_train_pool_20260307_163358/runs/metrics_latest.json"
],
"vocabulary": [
"models/shared/vocab.pkl"
],
"label_map_file": "models/shared/class2id.json",
"label_classes_file": "models/shared/label_classes.json",
"label_encoder_file": "models/shared/label_encoder.pkl"
},
"model": {
"embedding_dim": 300,
"hidden_dim": 128,
"num_layers": 2,
"dropout": 0.3,
"bidirectional": true,
"attention": "additive",
"vocab_size": 50000,
"pad_token": "<PAD>",
"unk_token": "<UNK>"
},
"runtime": {
"max_sequence_length": 256,
"truncation": true,
"padding": true,
"batch_size": 1,
"device": "cpu",
"preprocessing": "normalize_text + preprocess_from_normalized + texts_to_sequences"
},
"inference": {
"return_probabilities": true,
"confidence_method": "softmax_max"
},
"ui": {
"domain_display_name": "Text Complexity",
"color_token": "complexity",
"group": "complexity-core"
},
"labels": {
"type": "single-label-classification",
"classes": [
{
"id": 0,
"name": "A1-A2",
"display_name": "A1-A2"
},
{
"id": 1,
"name": "B1-B2",
"display_name": "B1-B2"
},
{
"id": 2,
"name": "C1-C2",
"display_name": "C1-C2"
}
]
},
"prod_model_dir": "app/app-models/prod-model-complexity",
"artifacts_present": [
"app/app-models/prod-model-complexity/.DS_Store",
"app/app-models/prod-model-complexity/best.pt",
"app/app-models/prod-model-complexity/label_classes.json",
"app/app-models/prod-model-complexity/model-config.yaml",
"app/app-models/prod-model-complexity/test_inference.ipynb",
"app/app-models/prod-model-complexity/vocab.pkl"
]
}