Bugfix: the predicted uncertainty is not outputed for external test set.

Xiangyan93 · Mar 6, 2024 · c6e0a85 · c6e0a85
1 parent 47ecfa2
commit c6e0a85
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mgktools/__init__.py b/mgktools/__init__.py
@@ -2,4 +2,4 @@
 # -*- coding: utf-8 -*-
 
 
-__version__ = '1.1.0'
+__version__ = '1.1.1'
diff --git a/mgktools/evaluators/cross_validation.py b/mgktools/evaluators/cross_validation.py
@@ -235,6 +235,8 @@ def predict(self, X, y: np.ndarray, repr: List[str], y_similar: List[str] = None
                 'predict': y_pred,
                 'repr': repr,
             }
+            if y_std is not None:
+                pred_dict['uncertainty'] = y_std
         elif y.ndim == 2:
             assert y_pred.ndim == 2
             assert y_std is None or y_std.ndim == 2
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,4 +2,4 @@
		# -- coding: utf-8 --


		__version__ = '1.1.0'
		__version__ = '1.1.1'