diff --git a/README.md b/README.md index 7ca3079..67be634 100644 --- a/README.md +++ b/README.md @@ -79,9 +79,17 @@ We are dealing with an exteremly imbalance dataset related to electrocardiogram model.fit(X_train, y_train) ``` +### STEP 4: Train different models to find the best possible approach + +What we are looking for: +Dangerous: Sick → predicted healthy : high recall score or low FN +Costly: Healthy → predicted sick : high precision score or low FP + + ``` -✅ 1. Stratified K-fold only apply on train. Of each fold +next steps: +✅ 1. Stratified K-fold only apply on train. 🗹 2. train LGBM model using KMEANS_SMOTE with k_neighbors=10 🗹 3. train Cat_boost using KMEANS_SMOTE with k_neighbors=10 🗹 4. implement proposed methods of this article : https://1drv.ms/b/c/ab2a38fe5c318317/IQBEDsSFcYj6R6AMtOnh0X6DAZUlFqAYq19WT8nTeXomFwg