Update README.md

README.md

@@ -45,7 +45,29 @@ probs = infer(input_texts)
 
 ## Model Performance
 
-Coming soon...
+The following table shows the results for each class on the validation data (n=150) of CT-RATE-JPN. For metrics requiring binarization, a threshold of 0.5 was used.
+
+
+|    | Disease                            |   Positive_Samples |   Accuracy |   Precision |   Recall |    F1 |   AUC-ROC |    AP |
+|---:|:-----------------------------------|-------------------:|-----------:|------------:|---------:|------:|----------:|------:|
+|  0 | Medical material                   |                 14 |      0.973 |       0.778 |    1     | 0.875 |     0.999 | 0.99  |
+|  1 | Arterial wall calcification        |                 49 |      0.987 |       0.961 |    1     | 0.98  |     1     | 1     |
+|  2 | Cardiomegaly                       |                 25 |      0.987 |       1     |    0.92  | 0.958 |     0.999 | 0.996 |
+|  3 | Pericardial effusion               |                 12 |      1     |       1     |    1     | 1     |     1     | 1     |
+|  4 | Coronary artery wall calcification |                 45 |      0.987 |       0.978 |    0.978 | 0.978 |     1     | 1     |
+|  5 | Hiatal hernia                      |                 24 |      1     |       1     |    1     | 1     |     1     | 1     |
+|  6 | Lymphadenopathy                    |                 37 |      0.987 |       0.973 |    0.973 | 0.973 |     0.994 | 0.987 |
+|  7 | Emphysema                          |                 31 |      0.98  |       0.938 |    0.968 | 0.952 |     0.989 | 0.96  |
+|  8 | Atelectasis                        |                 49 |      0.993 |       0.98  |    1     | 0.99  |     1     | 1     |
+|  9 | Lung nodule                        |                 82 |      0.967 |       0.975 |    0.963 | 0.969 |     0.991 | 0.994 |
+| 10 | Lung opacity                       |                 55 |      0.953 |       0.929 |    0.945 | 0.937 |     0.991 | 0.985 |
+| 11 | Pulmonary fibrotic sequela         |                 47 |      0.953 |       0.935 |    0.915 | 0.925 |     0.981 | 0.973 |
+| 12 | Pleural effusion                   |                 19 |      0.987 |       0.905 |    1     | 0.95  |     1     | 0.997 |
+| 13 | Mosaic attenuation pattern         |                 25 |      1     |       1     |    1     | 1     |     1     | 1     |
+| 14 | Peribronchial thickening           |                 21 |      0.96  |       1     |    0.714 | 0.833 |     0.985 | 0.948 |
+| 15 | Consolidation                      |                 24 |      0.933 |       0.706 |    1     | 0.828 |     0.996 | 0.985 |
+| 16 | Bronchiectasis                     |                 20 |      0.98  |       0.87  |    1     | 0.93  |     0.99  | 0.873 |
+| 17 | Interlobular septal thickening     |                  7 |      0.993 |       0.875 |    1     | 0.933 |     1     | 1     |
 
 ## Citation