Update README.md
Browse files
README.md
CHANGED
@@ -29,14 +29,14 @@ The model has not undergone clinical testing and usage is at ueser's own risk.T
|
|
29 |
|
30 |
#How to use
|
31 |
Download the pre-trained model and use it to make inference.
|
32 |
-
A space has been created for testing (
|
33 |
|
34 |
#Training data
|
35 |
The entire dataset consist of 3500 negative images and 700 positive TB images. </br>
|
36 |
The data was splitted in 80% for training and 20% for validation.
|
37 |
|
38 |
#Training procedure
|
39 |
-
Transfer-learning was employed using InceptionV3 as the pre-trained model. Training was done for 20 epochs and the classes were weighted during training in order to neutralize the imbalanced class in the dataset. The training was done on Kaggle using the GPUs provided. More details of the experiments can be found (
|
40 |
|
41 |
#Evaluation results
|
42 |
The result of the evaluation are as follows: - loss: 0.0923 - binary_accuracy: 0.9857 - precision: 0.9259 - recall: 0.9843
|
|
|
29 |
|
30 |
#How to use
|
31 |
Download the pre-trained model and use it to make inference.
|
32 |
+
A space has been created for testing [here](space.com)
|
33 |
|
34 |
#Training data
|
35 |
The entire dataset consist of 3500 negative images and 700 positive TB images. </br>
|
36 |
The data was splitted in 80% for training and 20% for validation.
|
37 |
|
38 |
#Training procedure
|
39 |
+
Transfer-learning was employed using InceptionV3 as the pre-trained model. Training was done for 20 epochs and the classes were weighted during training in order to neutralize the imbalanced class in the dataset. The training was done on Kaggle using the GPUs provided. More details of the experiments can be found [here](https://www.kaggle.com/code/abrahamowodunni/tb-project)
|
40 |
|
41 |
#Evaluation results
|
42 |
The result of the evaluation are as follows: - loss: 0.0923 - binary_accuracy: 0.9857 - precision: 0.9259 - recall: 0.9843
|