Upload folder using huggingface_hub

README.md

@@ -5,16 +5,18 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: Small pleural effusion on the left side.  Suspicious for malignancy; urgent
-    CT recommended.
-- text: Hyperinflated lungs, consistent with COPD.  No acute cardiopulmonary abnormalities.
-- text: Lungs clear, no consolidation, no tuberculosis. Normal mediastinal contour.
-    Hilar structures normal. No osseous lesions. nan Normal chest radiograph. No evidence
-    of infectious process. No follow-up needed. Suitable for immigration clearance.
-- text: Small pleural effusion on the left side.  Normal chest X-ray, no acute findings.
-- text: Heart size and mediastinal contours normal. Lungs clear, no focal consolidation.
-    0.8 cm non-calcified nodule in left upper lobe. No lymphadenopathy. No acute cardiopulmonary
-    abnormalities. Follow-up CT recommended. Chest CT recommended to assess nodule.
+- text: Multiple non-calcified nodules in left lower lobe, ranging from 3-6mm. Follow-up
+    CT scan recommended.
+- text: Heart size and mediastinal contours normal. Lungs clear, no consolidation
+    or acute abnormality. No pleural effusion or pneumothorax. No acute osseous abnormalities.
+    Small 0.5 cm calcified granuloma in right upper lobe, likely benign from prior
+    infection. No acute cardiopulmonary disease. Incidental right upper lobe granuloma.
+    No follow-up needed unless clinically indicated. nan
+- text: Patchy consolidation in RLL consistent with pneumonia. Recommend clinical
+    follow-up.
+- text: Calcified granuloma in the left lower lobe, incidental finding.  Findings
+    consistent with COPD exacerbation.
+- text: Moderate pleural effusion and patchy consolidation in bilateral lung bases.
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -67,7 +69,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("setfit_model_id")
 # Run inference
-preds = model("Small pleural effusion on the left side.  Normal chest X-ray, no acute findings.")
+preds = model("Moderate pleural effusion and patchy consolidation in bilateral lung bases.")
 ```
 
 <!--
@@ -99,7 +101,7 @@ preds = model("Small pleural effusion on the left side.  Normal chest X-ray, no
 ### Training Set Metrics
 | Training set | Min | Median  | Max |
 |:-------------|:----|:--------|:----|
-| Word count   | 11  | 24.3913 | 53  |
+| Word count   | 7   | 17.5143 | 53  |
 
 ### Training Hyperparameters
 - batch_size: (8, 8)
@@ -122,15 +124,27 @@ preds = model("Small pleural effusion on the left side.  Normal chest X-ray, no
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0244 | 1    | 0.1229        | -               |
-| 1.2195 | 50   | 0.1709        | -               |
-| 2.4390 | 100  | 0.0923        | -               |
-| 3.6585 | 150  | 0.0738        | -               |
-| 4.8780 | 200  | 0.073         | -               |
-| 6.0976 | 250  | 0.0634        | -               |
-| 7.3171 | 300  | 0.0627        | -               |
-| 8.5366 | 350  | 0.0555        | -               |
-| 9.7561 | 400  | 0.0583        | -               |
+| 0.0096 | 1    | 0.1417        | -               |
+| 0.4808 | 50   | 0.1824        | -               |
+| 0.9615 | 100  | 0.1083        | -               |
+| 1.4423 | 150  | 0.0817        | -               |
+| 1.9231 | 200  | 0.0777        | -               |
+| 2.4038 | 250  | 0.0636        | -               |
+| 2.8846 | 300  | 0.0649        | -               |
+| 3.3654 | 350  | 0.0603        | -               |
+| 3.8462 | 400  | 0.0713        | -               |
+| 4.3269 | 450  | 0.0507        | -               |
+| 4.8077 | 500  | 0.0569        | -               |
+| 5.2885 | 550  | 0.0553        | -               |
+| 5.7692 | 600  | 0.0614        | -               |
+| 6.25   | 650  | 0.0512        | -               |
+| 6.7308 | 700  | 0.0559        | -               |
+| 7.2115 | 750  | 0.0512        | -               |
+| 7.6923 | 800  | 0.0464        | -               |
+| 8.1731 | 850  | 0.0547        | -               |
+| 8.6538 | 900  | 0.0455        | -               |
+| 9.1346 | 950  | 0.0524        | -               |
+| 9.6154 | 1000 | 0.0526        | -               |
 
 ### Framework Versions
 - Python: 3.11.11

id2label.json

@@ -0,0 +1 @@
+{"0": "air bronchograms", "1": "annual screening", "2": "benign", "3": "calcification", "4": "calcified nodule", "5": "cardiomegaly", "6": "chest ct", "7": "clinical correlation", "8": "clinical monitoring", "9": "consolidation", "10": "copd", "11": "ct recommended", "12": "ct scan", "13": "effusion", "14": "follow-up", "15": "follow-uproutine follow-up", "16": "granuloma", "17": "hear size normal", "18": "heart size normal", "19": "hyperinflation", "20": "infection", "21": "infiltrates", "22": "large heart", "23": "large nodule", "24": "lungs clear", "25": "malignancy", "26": "no abnormalities", "27": "no acute findings", "28": "no acute issues", "29": "no calcification", "30": "no calcifications", "31": "no consolidation", "32": "no contraindications", "33": "no effusion", "34": "no follow-up needed", "35": "no granuloma", "36": "no infection", "37": "no infiltrates", "38": "no lymphadenopathy", "39": "no mass", "40": "no nodules", "41": "no pheumothorax", "42": "no pleural effusion", "43": "no pleurla effusion", "44": "no pneumothorax", "45": "no tuberculosis", "46": "nodule", "47": "nodules", "48": "normal chest xray", "49": "opacity", "50": "orthopedic consult", "51": "pleural effusion", "52": "pneumonia", "53": "pneumothorax", "54": "pneumothorax resolved", "55": "previous imaging", "56": "prior infection", "57": "pulmonary nodule", "58": "repeat imaging", "59": "rib fracture", "60": "smoker", "61": "suspicious", "62": "thoracic spine", "63": "urgent"}

label2id.json

@@ -1 +1 @@
-{"annual screening": 0, "cardiomegaly": 1, "copd": 2, "ct recommended": 3, "malignancy": 4, "no follow-up needed": 5, "nodule": 6, "nodule (unspecified)": 7, "normal": 8, "opacity": 9, "pleural effusion": 10, "pneumonia": 11, "pneumothorax": 12, "repeat imaging": 13, "rib fracture": 14, "small nodules": 15, "suspicious nodules": 16}
+{"air bronchograms": 0, "annual screening": 1, "benign": 2, "calcification": 3, "calcified nodule": 4, "cardiomegaly": 5, "chest ct": 6, "clinical correlation": 7, "clinical monitoring": 8, "consolidation": 9, "copd": 10, "ct recommended": 11, "ct scan": 12, "effusion": 13, "follow-up": 14, "follow-uproutine follow-up": 15, "granuloma": 16, "hear size normal": 17, "heart size normal": 18, "hyperinflation": 19, "infection": 20, "infiltrates": 21, "large heart": 22, "large nodule": 23, "lungs clear": 24, "malignancy": 25, "no abnormalities": 26, "no acute findings": 27, "no acute issues": 28, "no calcification": 29, "no calcifications": 30, "no consolidation": 31, "no contraindications": 32, "no effusion": 33, "no follow-up needed": 34, "no granuloma": 35, "no infection": 36, "no infiltrates": 37, "no lymphadenopathy": 38, "no mass": 39, "no nodules": 40, "no pheumothorax": 41, "no pleural effusion": 42, "no pleurla effusion": 43, "no pneumothorax": 44, "no tuberculosis": 45, "nodule": 46, "nodules": 47, "normal chest xray": 48, "opacity": 49, "orthopedic consult": 50, "pleural effusion": 51, "pneumonia": 52, "pneumothorax": 53, "pneumothorax resolved": 54, "previous imaging": 55, "prior infection": 56, "pulmonary nodule": 57, "repeat imaging": 58, "rib fracture": 59, "smoker": 60, "suspicious": 61, "thoracic spine": 62, "urgent": 63}

label_to_category.json

@@ -0,0 +1 @@
+{"lungs clear": "present_findings", "normal chest xray": "present_findings", "heart size normal": "present_findings", "thoracic spine": "present_findings", "calcification": "present_findings", "prior infection": "present_findings", "benign": "present_findings", "granuloma": "nodules", "nodules": "present_findings", "no contraindications": "present_findings", "large heart": "present_findings", "cardiomegaly": "present_findings", "nodule": "nodules", "smoker": "present_findings", "previous imaging": "present_findings", "hear size normal": "present_findings", "no calcification": "absent_findings", "no pneumothorax": "absent_findings", "no pleurla effusion": "present_findings", "no lymphadenopathy": "absent_findings", "rib fracture": "present_findings", "pleural effusion": "present_findings", "opacity": "present_findings", "malignancy": "present_findings", "infection": "present_findings", "suspicious": "present_findings", "pneumonia": "present_findings", "copd": "present_findings", "infiltrates": "present_findings", "follow-up": "follow_up", "ct scan": "follow_up", "hyperinflation": "present_findings", "consolidation": "present_findings", "pneumothorax resolved": "present_findings", "calcified nodule": "present_findings", "effusion": "present_findings", "pneumothorax": "present_findings", "air bronchograms": "present_findings", "no consolidation": "absent_findings", "no pheumothorax": "absent_findings", "no pleural effusion": "absent_findings", "no calcifications": "absent_findings", "no tuberculosis": "absent_findings", "no infection": "absent_findings", "no infiltrates": "absent_findings", "no abnormalities": "absent_findings", "no effusion": "absent_findings", "no acute issues": "absent_findings", "no acute findings": "absent_findings", "no nodules": "absent_findings", "no granuloma": "absent_findings", "no mass": "absent_findings", "no follow-up needed": "follow_up", "annual screening": "follow_up", "follow-uproutine follow-up": "follow_up", "chest ct": "follow_up", "ct recommended": "follow_up", "urgent": "follow_up", "repeat imaging": "follow_up", "orthopedic consult": "follow_up", "clinical correlation": "follow_up", "clinical monitoring": "follow_up", "pulmonary nodule": "nodules", "large nodule": "nodules"}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bf117229fd57b49ff004bae84f9160232b17619ece79389abaedb1a35d9911c5
+oid sha256:f8b425d5f7174fe6bf2ec482c1bff0edb932f906fb7adab5c54460afd9d0da3a
 size 433263448

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4ae784e085cb211b565c8788c712cd5a4fa33f6b4b154c6a4ac87088499c1b15
-size 111172
+oid sha256:c5dc90c146c5b5fc4f81bca83c3b76af9fb32e4bcfd3430cc900b2f32e9ffd81
+size 351924