modified: auton_survival/experiments.py

new file: docs/models/.DS_Store
autonlab · Jun 28, 2022 · 8e99b90 · 8e99b90
1 parent 3ed86ce
commit 8e99b90
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/auton_survival/experiments.py b/auton_survival/experiments.py
@@ -176,7 +176,7 @@ def fit(self, features, outcomes, horizons, metric='ibs'):
                                            predictions=predictions,
                                            times=horizons,
                                            outcomes_train=outcomes.loc[self.folds!=fold])
-        fold_scores.append(score)
+        fold_scores.append(np.mean(score))
       hyper_param_scores.append(np.mean(fold_scores)) 
 
     if self.metric in ['ibs', 'brs']:

diff --git a/docs/models/.DS_Store b/docs/models/.DS_Store