Skip to content

Commit 588ffbf

Browse files
committed
hack for traceable model and obtaining classes
1 parent 869d078 commit 588ffbf

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

openml/runs/functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def _run_task_get_arffcontent(model, task, class_labels):
167167
traceable_model = get_traceble_model(model_fold)
168168
if traceable_model:
169169
arff_tracecontent.extend(_extract_arfftrace(traceable_model, rep_no, fold_no))
170-
model_classes = model_fold.best_estimator_.classes_
170+
model_classes = traceable_model.best_estimator_.classes_
171171
else:
172172
model_classes = model_fold.classes_
173173

0 commit comments

Comments
 (0)