pyfemtet 0.9.5__py3-none-any.whl → 1.0.0b0__py3-none-any.whl

pyfemtet/opt/optimizer/optuna_optimizer/__init__.py

@@ -0,0 +1,9 @@
+from ._optuna_optimizer import OptunaOptimizer
+from ._pof_botorch import PoFConfig, PoFBoTorchSampler, PartialOptimizeACQFConfig
+
+__all__ = [
+    'OptunaOptimizer',
+    'PoFConfig',
+    'PoFBoTorchSampler',
+    'PartialOptimizeACQFConfig',
+]

pyfemtet/opt/optimizer/optuna_optimizer/_optuna_attribute.py

@@ -0,0 +1,73 @@
+from __future__ import annotations
+
+from pyfemtet.opt.problem.problem import *
+from pyfemtet.opt.history import *
+from pyfemtet.opt.optimizer._base_optimizer import *
+
+
+class OptunaAttribute:
+    """Manage optuna user attribute
+
+    user attributes are:
+        sub_fidelity_name:
+            fidelity: ...
+            OBJECTIVE_ATTR_KEY: ...
+            PYFEMTET_STATE_ATTR_KEY: ...
+            CONSTRAINT_ATTR_KEY: ...
+
+    """
+
+    OBJECTIVE_KEY = 'internal_objective'
+    CONSTRAINT_KEY = 'constraint'
+    PYFEMTET_TRIAL_STATE_KEY = 'pyfemtet_trial_state'
+
+    sub_fidelity_name: str  # key
+    fidelity: Fidelity | None
+    v_values: tuple | None  # violation
+    y_values: tuple | None  # internal objective
+    pf_state: TrialState | None  # PyFemtet state
+
+    def __init__(self, opt: AbstractOptimizer):
+        self.sub_fidelity_name = opt.sub_fidelity_name
+        self.fidelity = None
+        self.v_values = None
+        self.y_values = None
+        self.pf_state = None
+
+    # noinspection PyPropertyDefinition
+    @classmethod
+    def main_fidelity_key(cls):
+        return MAIN_FIDELITY_NAME
+
+    @property
+    def key(self):
+        return self.sub_fidelity_name
+
+    @property
+    def value(self):
+        d = {}
+        if self.fidelity:
+            d.update({'fidelity': self.fidelity})
+        if self.y_values:
+            d.update({self.OBJECTIVE_KEY: self.y_values})
+        if self.v_values:
+            d.update({self.CONSTRAINT_KEY: self.v_values})
+        if self.pf_state:
+            d.update({self.PYFEMTET_TRIAL_STATE_KEY: self.pf_state})
+        return d
+
+    @staticmethod
+    def get_fidelity(optuna_attribute: OptunaAttribute):
+        return optuna_attribute.value['fidelity']
+
+    @staticmethod
+    def get_violation(optuna_attribute: OptunaAttribute):
+        return optuna_attribute.value[OptunaAttribute.CONSTRAINT_KEY]
+
+    @staticmethod
+    def get_violation_from_trial_attr(trial_attr: dict):  # value is OptunaAttribute.value
+        return trial_attr[OptunaAttribute.CONSTRAINT_KEY]
+
+    @staticmethod
+    def get_pf_state_from_trial_attr(trial_attr: dict):  # value is OptunaAttribute.value
+        return trial_attr[OptunaAttribute.PYFEMTET_TRIAL_STATE_KEY]