Skip to content

Commit 020f678

Browse files
committed
README.md
1 parent 169db5d commit 020f678

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

README.md

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -174,11 +174,15 @@ In addition to these exploratory analyses, I tested my hypothesis and answered k
174174
- I split the train dataset into training (80%) and evaluation (20%) sets.
175175
- To properly train and evaluate the models, I took into consideration the imbalanced nature of the traget variables/ class in the training set.
176176

177-
![Class imbalance in the Train Set](screenshots/imb_class.png)
177+
<p align="center">
178+
<img src=screenshots/imb_class.png alt="class_imbalance" width="600">
179+
</p>
178180

179181
- I used random oversampling to balance the target class. Addressing this class imbalance is important to ensure that the machine learning model can make accurate predictions for both classes and is not biased towards the majority class.
180182

181-
![Class Distribution After Balancing](screenshots/bal_class.png)
183+
<p align="center">
184+
<img src=screenshots/bal_class.png alt="balanced_class" width="600">
185+
</p>
182186

183187
- I also scaled the training and evaluation sets to ensure they were within the same range.
184188

@@ -202,7 +206,9 @@ vii. Support Vector Machine
202206

203207
The model performance was as follows:
204208

205-
![Alt text](screenshots/f1_score.png)
209+
<p align="center">
210+
<img src=screenshots/f1_score.png alt="f1_score" width="600">
211+
</p>
206212

207213
Based on the f1 score, the Gradient Boosting Classifier was the best performing model.
208214

0 commit comments

Comments
 (0)