Updating model card
Browse files
README.md
CHANGED
|
@@ -2,8 +2,50 @@
|
|
| 2 |
tags:
|
| 3 |
- model_hub_mixin
|
| 4 |
- pytorch_model_hub_mixin
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
tags:
|
| 3 |
- model_hub_mixin
|
| 4 |
- pytorch_model_hub_mixin
|
| 5 |
+
metrics:
|
| 6 |
+
- F1
|
| 7 |
+
- Precision
|
| 8 |
+
- Recall
|
| 9 |
+
- ROC-AUC
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# ARespiratory audio classification model
|
| 13 |
+
|
| 14 |
+
This model classifies respiratory audio recordings from the [ICBHI 2017 Challenge](https://bhichallenge.med.auth.gr/ICBHI_2017_Challenge) dataset into **crackles**, **wheezes**, **both**, or **none** (multi-label classification). It utilizes the [AST encoder](https://huggingface.co/MIT/ast-finetuned-audioset-14-14-0.443) (`MIT/ast-finetuned-audioset-14-14-0.443`) with a lightweight classification head.
|
| 15 |
+
|
| 16 |
+
The model has been pushed to the Hub using the [PyTorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration.
|
| 17 |
+
|
| 18 |
+
---
|
| 19 |
+
|
| 20 |
+
## **Dataset**
|
| 21 |
+
|
| 22 |
+
- **Source:** [ICBHI 2017 Challenge](https://bhichallenge.med.auth.gr/ICBHI_2017_Challenge)
|
| 23 |
+
|
| 24 |
+
---
|
| 25 |
+
|
| 26 |
+
## **Performance metrics**
|
| 27 |
+
|
| 28 |
+
| **Label** | **F1** | **Precision** | **Recall** | **AUC** |
|
| 29 |
+
|--------------|---------|---------------|------------|-----------|
|
| 30 |
+
| **Crackle** | 0.6756 | 0.6147 | 0.7500 | 0.7033 |
|
| 31 |
+
| **Wheeze** | 0.4853 | 0.6565 | 0.3849 | 0.8031 |
|
| 32 |
+
| **Macro Avg**| 0.5805 | 0.6356 | 0.5674 | 0.7532 |
|
| 33 |
+
|
| 34 |
+
---
|
| 35 |
+
|
| 36 |
+
## **Usage**
|
| 37 |
+
|
| 38 |
+
Run inference using [`s05_inference.py`](https://github.com/fabiocat93/icbhi_2017_challenge/blob/main/src/code/s05_inference.py).
|
| 39 |
+
Ensure you install the necessary dependencies. For setup instructions, please see the [documentation](https://github.com/fabiocat93/icbhi_2017_challenge).
|
| 40 |
+
|
| 41 |
+
---
|
| 42 |
+
|
| 43 |
+
## **Notes**
|
| 44 |
+
|
| 45 |
+
For additional details, check the [documentation notes](https://github.com/fabiocat93/icbhi_2017_challenge/edit/main/docs/notes.md).
|
| 46 |
+
|
| 47 |
+
---
|
| 48 |
+
|
| 49 |
+
## **Contact**
|
| 50 |
+
|
| 51 |
+
For any questions or further information, feel free to reach out via email: **[[email protected]](mailto:[email protected])**.
|