fabiocat commited on
Commit
d698733
·
verified ·
1 Parent(s): f29f4d7

Updating model card

Browse files
Files changed (1) hide show
  1. README.md +45 -3
README.md CHANGED
@@ -2,8 +2,50 @@
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
 
 
 
 
 
5
  ---
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Library: [More Information Needed]
9
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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])**.