Srivardhan369 commited on
Commit
6bf654b
·
verified ·
1 Parent(s): 53a2592

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -3
README.md CHANGED
@@ -1,3 +1,43 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets: brain-tumor-image-dataset-semantic-segmentation
4
+ metrics:
5
+ - accuracy
6
+ - f1
7
+ - precision
8
+ - recall
9
+ pipeline_tag: image-classification
10
+ tags:
11
+ - brain-tumor
12
+ - image-classification
13
+ - keras
14
+ - tensorflow
15
+ - cnn
16
+ - mri
17
+ - healthcare
18
+ ---
19
+
20
+ # Tumor Detection ML Model
21
+
22
+ ## Model Description
23
+ This model is designed to classify brain tumor images using a Convolutional Neural Network (CNN). It has been trained and fine-tuned on a labeled dataset of brain tumor MRI images.
24
+
25
+ ## Training Details
26
+ - **Framework:** TensorFlow/Keras
27
+ - **Optimizer:** Adam with a learning rate scheduler
28
+ - **Loss Function:** Categorical Crossentropy
29
+ - **Data Augmentation:** Includes rotation, width/height shift, zoom, and horizontal flipping.
30
+ - **Hyperparameter Tuning:** Performed using Keras Tuner.
31
+
32
+ ## Metrics
33
+ The following metrics were used to evaluate the model's performance:
34
+ - **Accuracy:** Measures the overall correctness of predictions.
35
+ - **F1 Score:** Balances precision and recall.
36
+ - **Precision:** Indicates the proportion of true positives among positive predictions.
37
+ - **Recall:** Indicates the proportion of true positives among all actual positives.
38
+
39
+ ## Usage
40
+ You can load the model using the Hugging Face Transformers library:
41
+ ```python
42
+ from transformers import AutoModel
43
+ model = AutoModel.from_pretrained("YourUsername/Tumor_detection_ML_Model")