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

adding predictors.pdf

Browse Source
This commit is contained in:
Saeed Khosravi
2025-12-10 21:38:28 +01:00
parent 0f1a05f293
commit ac86621ad2
6 changed files with 508 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
models/catboost_model.py
View File

@@ -72,11 +72,11 @@ class CAT_BOOST:
desc=" >> CatBoost Fitting: ",
)
):
x_train_fold, x_val = (
x_train_fold, x_val_fold = (
x_train_val.iloc[train_index],
x_train_val.iloc[val_index],
)
y_train_fold, y_val = (
y_train_fold, y_val_fold = (
y_train_val.iloc[train_index],
y_train_val.iloc[val_index],
)
@@ -114,8 +114,8 @@ class CAT_BOOST:
)
self.model.fit(x_train_fold, y_train_fold)
y_pred_val = self.model.predict(x_val)
val_metrics = get_metrics(y_val, y_pred_val)
y_pred_val = self.model.predict(x_val_fold)
val_metrics = get_metrics(y_val_fold, y_pred_val)
fold_results.append(val_metrics)
return average_fold_results(fold_results)