get mean

CyberAgentAILab · Dec 19, 2023 · ea11b1f · ea11b1f
1 parent 07d26c9
commit ea11b1f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cmaes/_cma.py b/cmaes/_cma.py
@@ -248,6 +248,11 @@ def generation(self) -> int:
         when multi-variate gaussian distribution is updated."""
         return self._g
 
+    @property
+    def mean(self) -> np.ndarray:
+        """Mean Vector"""
+        return self._mean
+
     def reseed_rng(self, seed: int) -> None:
         self._rng.seed(seed)
 

diff --git a/cmaes/_cmawm.py b/cmaes/_cmawm.py
@@ -183,6 +183,11 @@ def generation(self) -> int:
         when multi-variate gaussian distribution is updated."""
         return self._cma.generation
 
+    @property
+    def mean(self) -> np.ndarray:
+        """Mean Vector"""
+        return self._cma.mean
+
     @property
     def _rng(self) -> np.random.RandomState:
         return self._cma._rng