Merge pull request #10 from Techtonique/ucb-tol

use method='splitconformal' in predict
Techtonique · Dec 6, 2024 · edc4afc · edc4afc
2 parents bd10243 + 5877393
commit edc4afc
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/GPopt/GPOpt.py b/GPopt/GPOpt.py
@@ -340,7 +340,7 @@ def surrogate_fit_predict(
                 assert self.acquisition == "ucb", "'acquisition' must be 'ucb' for conformalized surrogates"
                 self.posterior_ = None 
                 res = self.surrogate_obj.fit(X_train, y_train).predict(
-                    X_test, return_pi=True)
+                    X_test, return_pi=True, method="splitconformal")
                 self.y_mean = res.mean
                 self.y_lower = res.lower 
                 self.y_upper = res.upper 

diff --git a/GPopt/demo/thierrymoudiki_20240206_tuning_BCN_classifier_Pt2.ipynb b/GPopt/demo/thierrymoudiki_20240206_tuning_BCN_classifier_Pt2.ipynb
@@ -258,7 +258,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.11.8"
+      "version": "3.11.10"
     }
   },
   "nbformat": 4,

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 from codecs import open
 from os import path
 
-__version__ = "0.8.0"
+__version__ = "0.8.1"
 
 subprocess.call("pip install -r requirements.txt", shell=True)