Upload folder using huggingface_hub

README.md

@@ -5,15 +5,16 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: Small pleural effusion on the left side.  Mild cardiomegaly; no acute issues.
-- text: Calcified granuloma in the left lower lobe, incidental finding.  Findings
-    consistent with COPD exacerbation.
-- text: Right upper lobe consolidation consistent with pneumonia.  Findings suggestive
-    of community-acquired pneumonia.
 - text: Small pleural effusion on the left side.  Suspicious for malignancy; urgent
     CT recommended.
-- text: Suspected rib fracture; follow-up imaging recommended.  Mild cardiomegaly;
-    no acute issues.
+- 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.
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -66,7 +67,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.  Mild cardiomegaly; no acute issues.")
+preds = model("Small pleural effusion on the left side.  Normal chest X-ray, no acute findings.")
 ```
 
 <!--
@@ -98,7 +99,7 @@ preds = model("Small pleural effusion on the left side.  Mild cardiomegaly; no a
 ### Training Set Metrics
 | Training set | Min | Median  | Max |
 |:-------------|:----|:--------|:----|
-| Word count   | 10  | 23.3913 | 53  |
+| Word count   | 11  | 24.3913 | 53  |
 
 ### Training Hyperparameters
 - batch_size: (8, 8)
@@ -121,15 +122,15 @@ preds = model("Small pleural effusion on the left side.  Mild cardiomegaly; no a
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0238 | 1    | 0.1381        | -               |
-| 1.1905 | 50   | 0.1559        | -               |
-| 2.3810 | 100  | 0.0962        | -               |
-| 3.5714 | 150  | 0.072         | -               |
-| 4.7619 | 200  | 0.068         | -               |
-| 5.9524 | 250  | 0.0652        | -               |
-| 7.1429 | 300  | 0.0574        | -               |
-| 8.3333 | 350  | 0.0579        | -               |
-| 9.5238 | 400  | 0.0589        | -               |
+| 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        | -               |
 
 ### Framework Versions
 - Python: 3.11.11

label2id.json

@@ -0,0 +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}

label_map.json

@@ -0,0 +1 @@
+{"0": "annual screening", "1": "cardiomegaly", "2": "copd", "3": "ct recommended", "4": "malignancy", "5": "no follow-up needed", "6": "nodule", "7": "nodule (unspecified)", "8": "normal", "9": "opacity", "10": "pleural effusion", "11": "pneumonia", "12": "pneumothorax", "13": "repeat imaging", "14": "rib fracture", "15": "small nodules", "16": "suspicious nodules"}

model.safetensors

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

model_head.pkl

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