1
0
Fork 0

fixing bug in ACC when using cross validation

This commit is contained in:
Pablo González 2022-06-24 14:20:08 +02:00
parent 02dd2846ff
commit 750814ef2a
1 changed files with 1 additions and 0 deletions

View File

@ -223,6 +223,7 @@ def cross_generate_predictions(
# fit the learner on all data # fit the learner on all data
learner.fit(*data.Xy) learner.fit(*data.Xy)
y = data.y
classes = data.classes_ classes = data.classes_
else: else:
learner, val_data = _training_helper( learner, val_data = _training_helper(