From 8acc99a42b8adaa267a334eb217fb0ad8cab7c48 Mon Sep 17 00:00:00 2001 From: Yuting Xu <12775874+xuyuting@users.noreply.github.com> Date: Thu, 15 Aug 2024 01:47:36 -0400 Subject: [PATCH] Update bayesian.py: revise 'score' attribute in surrogate to 'r2_score' to avoid conflict with instance method score() --- obsidian/optimizer/bayesian.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/obsidian/optimizer/bayesian.py b/obsidian/optimizer/bayesian.py index 7f8dfb0..343b26b 100644 --- a/obsidian/optimizer/bayesian.py +++ b/obsidian/optimizer/bayesian.py @@ -244,7 +244,7 @@ def fit(self, if self.verbose > 0: print(f'{self.surrogate_type[i]} model has been fit \ - to data with a train-score of: {self.surrogate[i].score:.3g} for response: {self.y_names[i]}') + to data with a train-score of: {self.surrogate[i].r2_score:.3g} for response: {self.y_names[i]}') return def save_state(self) -> dict: