Merge pull request #159 from CyberAgentAILab/feature-get-mean

get mean
CyberAgentAILab · Dec 27, 2023 · 057798d · 057798d
2 parents 07d26c9 + 22b4763
commit 057798d
Show file tree

Hide file tree

Showing 3 changed files with 15 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

diff --git a/cmaes/_sepcma.py b/cmaes/_sepcma.py
@@ -177,6 +177,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)