Skip to content

Commit 148e942

Browse files
committed
change output string in plot_eval
1 parent d6c0016 commit 148e942

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

autoemulate/compare.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -479,9 +479,7 @@ def evaluate(self, model=None, multioutput="uniform_average"):
479479

480480
scores_df = (
481481
pd.DataFrame(scores)
482-
.assign(
483-
target=[f"target_{i}" for i in range(len(scores[next(iter(scores))]))]
484-
)
482+
.assign(target=[f"y{i}" for i in range(len(scores[next(iter(scores))]))])
485483
.assign(short=get_short_model_name(model))
486484
.assign(model=get_model_name(model))
487485
.reindex(columns=["model", "short", "target"] + list(scores.keys()))

0 commit comments

Comments
 (0)