diff --git a/mgktools/__init__.py b/mgktools/__init__.py index 7c0640f..1af1c6b 100644 --- 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 index 71dacba..783f062 100644 --- 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