editing readme.md

2025-12-01 00:10:33 +01:00
parent 1a2e9cfb95
commit f6a86cfbe9
9 changed files with 1789 additions and 237 deletions
--- a/train.py
+++ b/train.py
@@ -43,7 +43,7 @@ def train_model_with_kfold(
        if smote:
            if smote_method.lower() == "kmeans":
                sampler = KMeansSMOTE(
-                    k_neighbors=5,
+                    k_neighbors=10,
                    cluster_balance_threshold=0.1,
                    random_state=random_state,
                )