Skip to content

Commit 3b421db

Browse files
committed
update fit_cross_validation
1 parent 492c384 commit 3b421db

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deeptables/models/hyper_dt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def fit_cross_validation(self, X, y, eval_set=None, metrics=None, pos_label=None
176176

177177
self.classes_ = getattr(self.model, 'classes_', None)
178178

179-
return scores, oof_proba, oof_scores, None, None # TODO: add xvals, yvals
179+
return scores, oof_proba, oof_scores, None, None, None, None
180180

181181
def predict(self, X, **kwargs):
182182
return self.model.predict(X, **kwargs)

0 commit comments

Comments
 (0)