saeed/Electrocardiogram
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

editing readme.md

Browse Source
This commit is contained in:
Saeed Khosravi
2025-12-01 00:34:33 +01:00
parent f6a86cfbe9
commit ad01d6a3db
8 changed files with 1021 additions and 1013 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
train.py
View File

@@ -42,19 +42,24 @@ def train_model_with_kfold(
if smote:
if smote_method.lower() == "kmeans":
from collections import Counter
minority = Counter(y_train)[1]
k_neighbors = min(10, max(2, minority // 10))
sampler = KMeansSMOTE(
k_neighbors=10,
k_neighbors=k_neighbors,
cluster_balance_threshold=0.1,
random_state=random_state,
)
elif smote_method.lower() == "smote":
sampler = SMOTE(k_neighbors=5, random_state=random_state)
sampler = SMOTE(k_neighbors=15, random_state=random_state)
elif smote_method.lower() == "svmsmote":
sampler = SVMSMOTE(k_neighbors=5, random_state=random_state)
sampler = SVMSMOTE(k_neighbors=15, random_state=random_state)
elif smote_method.lower() == "borderline":
sampler = BorderlineSMOTE(k_neighbors=5, random_state=random_state)
sampler = BorderlineSMOTE(k_neighbors=15, random_state=random_state)
elif smote_method.lower() == "adasyn":
sampler = ADASYN(n_neighbors=5, random_state=random_state)
sampler = ADASYN(n_neighbors=15, random_state=random_state)
else:
raise ValueError(f"Unknown smote_method: {smote_method}")