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])**.
|