Text Classification
Transformers
Safetensors
English
chest2vec_labeler
feature-extraction
radiology
chest-ct
report-labeling
multi-label
ct-rate
chexbert-style-f1
custom_code
Instructions to use chest2vec/chest2vec_labeler with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use chest2vec/chest2vec_labeler with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="chest2vec/chest2vec_labeler", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("chest2vec/chest2vec_labeler", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Upload config.json with huggingface_hub
Browse files- config.json +228 -1
config.json
CHANGED
|
@@ -269,5 +269,232 @@
|
|
| 269 |
"default_threshold": 0.5,
|
| 270 |
"torch_dtype": "float32",
|
| 271 |
"n_labels": 137,
|
| 272 |
-
"attn_implementation": "sdpa"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 273 |
}
|
|
|
|
| 269 |
"default_threshold": 0.5,
|
| 270 |
"torch_dtype": "float32",
|
| 271 |
"n_labels": 137,
|
| 272 |
+
"attn_implementation": "sdpa",
|
| 273 |
+
"label_hierarchy": {
|
| 274 |
+
"Lungs & Airways": {
|
| 275 |
+
"Lung parenchymal attenuation patterns": [
|
| 276 |
+
"Consolidation",
|
| 277 |
+
"Ground-glass opacity (GGO)",
|
| 278 |
+
"Crazy-paving pattern",
|
| 279 |
+
"Mosaic attenuation / air-trapping",
|
| 280 |
+
"Tree-in-bud",
|
| 281 |
+
"Centrilobular nodules / bronchiolitis pattern"
|
| 282 |
+
],
|
| 283 |
+
"Nodules and masses": [
|
| 284 |
+
"Pulmonary nodule (solid / PSN / GGN)",
|
| 285 |
+
"Pulmonary mass (>3 cm)",
|
| 286 |
+
"Cavitary nodule / mass"
|
| 287 |
+
],
|
| 288 |
+
"Emphysema & cystic lung disease": [
|
| 289 |
+
"Emphysema",
|
| 290 |
+
"Bullae / giant bulla",
|
| 291 |
+
"Pulmonary cysts / cystic lung disease"
|
| 292 |
+
],
|
| 293 |
+
"Interstitial / fibrotic lung disease": [
|
| 294 |
+
"Reticulation / intralobular thickening",
|
| 295 |
+
"Interlobular septal thickening",
|
| 296 |
+
"Traction bronchiectasis / bronchiolectasis",
|
| 297 |
+
"Honeycombing",
|
| 298 |
+
"Parenchymal scarring / fibrotic band"
|
| 299 |
+
],
|
| 300 |
+
"Volume loss / hyperinflation": [
|
| 301 |
+
"Lobar / segmental atelectasis",
|
| 302 |
+
"Subsegmental / linear atelectasis"
|
| 303 |
+
]
|
| 304 |
+
},
|
| 305 |
+
"Trachea & Airways": {
|
| 306 |
+
"Trachea & Airways": [
|
| 307 |
+
"Tracheal stenosis / malacia",
|
| 308 |
+
"Tracheal / bronchial wall thickening",
|
| 309 |
+
"Bronchiectasis",
|
| 310 |
+
"Mucoid impaction / plugging",
|
| 311 |
+
"Tracheal diverticulum"
|
| 312 |
+
]
|
| 313 |
+
},
|
| 314 |
+
"Pleura": {
|
| 315 |
+
"Pleural fluid collections": [
|
| 316 |
+
"Pleural effusion",
|
| 317 |
+
"Loculated pleural effusion",
|
| 318 |
+
"Hemothorax"
|
| 319 |
+
],
|
| 320 |
+
"Pleural air": [
|
| 321 |
+
"Pneumothorax",
|
| 322 |
+
"Tension pneumothorax"
|
| 323 |
+
],
|
| 324 |
+
"Pleural thickening / masses": [
|
| 325 |
+
"Pleural thickening",
|
| 326 |
+
"Pleural plaques",
|
| 327 |
+
"Pleural nodule / mass"
|
| 328 |
+
]
|
| 329 |
+
},
|
| 330 |
+
"Mediastinum & Hila": {
|
| 331 |
+
"Lymph nodes (mediastinal)": [
|
| 332 |
+
"Mediastinal lymphadenopathy",
|
| 333 |
+
"Hilar lymphadenopathy",
|
| 334 |
+
"Calcified mediastinal / hilar lymph nodes"
|
| 335 |
+
],
|
| 336 |
+
"Mediastinal masses / cysts": [
|
| 337 |
+
"Anterior mediastinal mass",
|
| 338 |
+
"Middle / posterior mediastinal mass or cyst",
|
| 339 |
+
"Thymic remnant / hyperplasia"
|
| 340 |
+
],
|
| 341 |
+
"Esophagus / hiatus": [
|
| 342 |
+
"Esophageal wall thickening / mass",
|
| 343 |
+
"Hiatal hernia",
|
| 344 |
+
"Esophageal dilation"
|
| 345 |
+
],
|
| 346 |
+
"Air / fluid in mediastinum": [
|
| 347 |
+
"Pneumomediastinum",
|
| 348 |
+
"Mediastinal hematoma / fluid collection"
|
| 349 |
+
]
|
| 350 |
+
},
|
| 351 |
+
"Cardiovascular": {
|
| 352 |
+
"Cardiac size and morphology": [
|
| 353 |
+
"Cardiomegaly"
|
| 354 |
+
],
|
| 355 |
+
"Pericardium": [
|
| 356 |
+
"Pericardial effusion",
|
| 357 |
+
"Pericardial thickening / calcification"
|
| 358 |
+
],
|
| 359 |
+
"Coronary arteries": [
|
| 360 |
+
"Coronary artery calcification",
|
| 361 |
+
"Coronary stent or bypass graft"
|
| 362 |
+
],
|
| 363 |
+
"Aorta / pulmonary artery": [
|
| 364 |
+
"Thoracic aortic calcification",
|
| 365 |
+
"Thoracic aortic ectasia / dilation (non-aneurysmal)",
|
| 366 |
+
"Thoracic aortic aneurysm",
|
| 367 |
+
"Aortic dissection / intramural hematoma",
|
| 368 |
+
"Main pulmonary artery enlargement",
|
| 369 |
+
"Pulmonary embolism"
|
| 370 |
+
],
|
| 371 |
+
"Valves and cardiac devices": [
|
| 372 |
+
"Aortic valve calcification",
|
| 373 |
+
"Mitral annular calcification",
|
| 374 |
+
"Pacemaker / ICD leads"
|
| 375 |
+
]
|
| 376 |
+
},
|
| 377 |
+
"Chest Wall": {
|
| 378 |
+
"Chest wall soft tissues": [
|
| 379 |
+
"Chest wall soft tissue edema / hematoma",
|
| 380 |
+
"Subcutaneous emphysema",
|
| 381 |
+
"Chest wall mass"
|
| 382 |
+
],
|
| 383 |
+
"Post-surgical / tumor invasion (chest wall)": [
|
| 384 |
+
"Post-thoracotomy change",
|
| 385 |
+
"Post-mastectomy change",
|
| 386 |
+
"Chest wall tumor invasion"
|
| 387 |
+
]
|
| 388 |
+
},
|
| 389 |
+
"Bones / Spine": {
|
| 390 |
+
"Fractures": [
|
| 391 |
+
"Acute rib fracture",
|
| 392 |
+
"Non-acute / healed rib fracture",
|
| 393 |
+
"Sternal fracture",
|
| 394 |
+
"Vertebral compression fracture"
|
| 395 |
+
],
|
| 396 |
+
"Degenerative / osseous lesions": [
|
| 397 |
+
"Degenerative spine changes",
|
| 398 |
+
"Osteolytic bone lesion",
|
| 399 |
+
"Osteosclerotic bone lesion",
|
| 400 |
+
"Mixed osteolytic-osteosclerotic lesion",
|
| 401 |
+
"Osteopenia",
|
| 402 |
+
"Scoliosis / kyphosis",
|
| 403 |
+
"Vertebral hemangioma",
|
| 404 |
+
"Postoperative spine change / hardware"
|
| 405 |
+
]
|
| 406 |
+
},
|
| 407 |
+
"Upper Abdomen": {
|
| 408 |
+
"Liver": [
|
| 409 |
+
"Hepatic steatosis",
|
| 410 |
+
"Focal liver lesion (nodule / mass)",
|
| 411 |
+
"Hepatomegaly",
|
| 412 |
+
"Liver contour irregularity / cirrhosis features",
|
| 413 |
+
"Hepatic calcification"
|
| 414 |
+
],
|
| 415 |
+
"Gallbladder & biliary": [
|
| 416 |
+
"Cholelithiasis / gallstones",
|
| 417 |
+
"Post-cholecystectomy (gallbladder operated / absent)",
|
| 418 |
+
"Gallbladder wall thickening",
|
| 419 |
+
"Hydropic gallbladder / distension",
|
| 420 |
+
"Biliary sludge",
|
| 421 |
+
"Biliary stent / catheter / drain"
|
| 422 |
+
],
|
| 423 |
+
"Spleen": [
|
| 424 |
+
"Splenomegaly",
|
| 425 |
+
"Accessory spleen / splenule / polysplenia",
|
| 426 |
+
"Focal splenic lesion (nodule / mass)"
|
| 427 |
+
],
|
| 428 |
+
"Pancreas": [
|
| 429 |
+
"Pancreatic mass / focal lesion",
|
| 430 |
+
"Pancreatic lipomatosis"
|
| 431 |
+
],
|
| 432 |
+
"Adrenals": [
|
| 433 |
+
"Adrenal nodule / mass",
|
| 434 |
+
"Adrenal thickening / hyperplasia",
|
| 435 |
+
"Adrenal calcification"
|
| 436 |
+
],
|
| 437 |
+
"Kidneys / urinary tract": [
|
| 438 |
+
"Simple renal cyst",
|
| 439 |
+
"Complex renal cyst / solid renal mass",
|
| 440 |
+
"Hydronephrosis",
|
| 441 |
+
"Renal calculi / nephrolithiasis",
|
| 442 |
+
"Renal atrophy / decreased renal size",
|
| 443 |
+
"Nephrectomy (kidney absent / operated)"
|
| 444 |
+
],
|
| 445 |
+
"GI & peritoneum": [
|
| 446 |
+
"Ascites",
|
| 447 |
+
"Pneumoperitoneum",
|
| 448 |
+
"Bowel wall thickening / inflammation",
|
| 449 |
+
"Diverticulosis",
|
| 450 |
+
"Omental caking / peritoneal carcinomatosis"
|
| 451 |
+
],
|
| 452 |
+
"Lymph nodes (abdominal)": [
|
| 453 |
+
"Abdominal lymphadenopathy"
|
| 454 |
+
],
|
| 455 |
+
"Upper abdominal vessels": [
|
| 456 |
+
"Abdominal aortic aneurysm (partially imaged)",
|
| 457 |
+
"Abdominal aortic calcification / atherosclerosis (partially imaged)"
|
| 458 |
+
]
|
| 459 |
+
},
|
| 460 |
+
"Lower Neck": {
|
| 461 |
+
"Thyroid": [
|
| 462 |
+
"Thyroid enlargement (goiter)",
|
| 463 |
+
"Thyroid nodule"
|
| 464 |
+
],
|
| 465 |
+
"Neck nodes and soft tissues": [
|
| 466 |
+
"Cervical / supraclavicular lymphadenopathy",
|
| 467 |
+
"Neck soft tissue mass"
|
| 468 |
+
]
|
| 469 |
+
},
|
| 470 |
+
"Others": {
|
| 471 |
+
"Breast": [
|
| 472 |
+
"Breast mass / focal asymmetry",
|
| 473 |
+
"Post-lumpectomy / post-mastectomy change",
|
| 474 |
+
"Breast implant (intact or present)"
|
| 475 |
+
],
|
| 476 |
+
"Axilla": [
|
| 477 |
+
"Axillary lymphadenopathy"
|
| 478 |
+
],
|
| 479 |
+
"Tubes, catheters, and support devices": [
|
| 480 |
+
"Endotracheal tube",
|
| 481 |
+
"Tracheostomy tube",
|
| 482 |
+
"Central venous catheter / PICC",
|
| 483 |
+
"Chest tube / pleural drain",
|
| 484 |
+
"Nasogastric / orogastric tube",
|
| 485 |
+
"IVC filter",
|
| 486 |
+
"LVAD / other cardiac assist device"
|
| 487 |
+
],
|
| 488 |
+
"Post-surgical / transplant status": [
|
| 489 |
+
"Post-lobectomy / segmentectomy",
|
| 490 |
+
"Post-pneumonectomy",
|
| 491 |
+
"Lung transplant"
|
| 492 |
+
],
|
| 493 |
+
"Study quality / global": [
|
| 494 |
+
"Motion artifact / suboptimal study",
|
| 495 |
+
"Study limitation / limited evaluation (non-motion)",
|
| 496 |
+
"No significant intrathoracic abnormality"
|
| 497 |
+
]
|
| 498 |
+
}
|
| 499 |
+
}
|
| 500 |
}
|