pyfemtet 0.7.1__py3-none-any.whl → 0.8.1__py3-none-any.whl

pyfemtet/opt/advanced_samples/surrogate_model/gal_ex13_create_training_data.py

@@ -0,0 +1,60 @@
+import os
+from time import sleep
+
+from optuna.samplers import RandomSampler
+
+from pyfemtet.opt import FEMOpt, FemtetInterface, OptunaOptimizer
+
+
+def get_res_freq(Femtet):
+    Galileo = Femtet.Gogh.Galileo
+    Galileo.Mode = 0
+    sleep(0.01)
+    return Galileo.GetFreq().Real
+
+
+if __name__ == '__main__':
+
+    os.chdir(os.path.dirname(__file__))
+
+    # Connect to Femtet.
+    fem = FemtetInterface(
+        femprj_path='gal_ex13_parametric.femprj',
+    )
+
+    # Initialize the optimization object.
+    # However, this script is not for optimization;
+    # instead, it is for creating training data.
+    # Therefore, we will use Optuna's random sampling
+    # class to select the design variables.
+    opt = OptunaOptimizer(
+        sampler_class=RandomSampler,
+    )
+
+    # We will set up the FEMOpt object. To refer to
+    # history_path in the optimization script, we will
+    # specify a clear CSV file name.
+    femopt = FEMOpt(
+        fem=fem,
+        opt=opt,
+        history_path='training_data.csv'
+    )
+
+    # Set the design variables.
+    femopt.add_parameter('length', 0.1, 0.02, 0.2)
+    femopt.add_parameter('width', 0.01, 0.001, 0.02)
+    femopt.add_parameter('base_radius', 0.008, 0.006, 0.01)
+
+    # Set the objective function. Since this is random
+    # sampling, specifying the direction does not affect
+    # the sampling.
+    femopt.add_objective(fun=get_res_freq, name='First Resonant Frequency (Hz)')
+
+    # Create the training data.
+    # If no termination condition is specified,
+    # it will continue creating training data until
+    # manually stopped.
+    femopt.set_random_seed(42)
+    femopt.optimize(
+        # n_trials=100
+    )

pyfemtet/opt/advanced_samples/surrogate_model/gal_ex13_create_training_data_jp.py

@@ -0,0 +1,57 @@
+import os
+from time import sleep
+
+from optuna.samplers import RandomSampler
+
+from pyfemtet.opt import FEMOpt, FemtetInterface, OptunaOptimizer
+
+
+def get_res_freq(Femtet):
+    Galileo = Femtet.Gogh.Galileo
+    Galileo.Mode = 0
+    sleep(0.01)
+    return Galileo.GetFreq().Real
+
+
+if __name__ == '__main__':
+
+    os.chdir(os.path.dirname(__file__))
+
+    # Femtet との接続を行います。
+    fem = FemtetInterface(
+        femprj_path='gal_ex13_parametric.femprj',
+    )
+
+    # 最適化用オブジェクトの設定を行います。
+    # ただしこのスクリプトでは最適化ではなく
+    # 学習データ作成を行うので、 optuna の
+    # ランダムサンプリングクラスを用いて
+    # 設計変数の選定を行います。
+    opt = OptunaOptimizer(
+        sampler_class=RandomSampler,
+    )
+
+    # FEMOpt オブジェクトを設定します。
+    # 最適化スクリプトで history_path を参照するため、
+    # わかりやすい csv ファイル名を指定します。
+    femopt = FEMOpt(
+        fem=fem,
+        opt=opt,
+        history_path='training_data.csv'
+    )
+
+    # 設計変数を設定します。
+    femopt.add_parameter('length', 0.1, 0.02, 0.2)
+    femopt.add_parameter('width', 0.01, 0.001, 0.02)
+    femopt.add_parameter('base_radius', 0.008, 0.006, 0.01)
+    # 目的関数を設定します。ランダムサンプリングなので
+    # direction は指定してもサンプリングに影響しません。
+    femopt.add_objective(fun=get_res_freq, name='第一共振周波数(Hz)')
+
+    # 学習データ作成を行います。
+    # 終了条件を指定しない場合、手動で停止するまで
+    # 学習データ作成を続けます。
+    femopt.set_random_seed(42)
+    femopt.optimize(
+        # n_trials=100
+    )

pyfemtet/opt/advanced_samples/surrogate_model/gal_ex13_optimize_with_surrogate.py

@@ -0,0 +1,100 @@
+import os
+
+from optuna.samplers import TPESampler
+
+from pyfemtet.opt import FEMOpt, OptunaOptimizer
+from pyfemtet.opt.interface import PoFBoTorchInterface
+
+
+def main(target):
+
+    os.chdir(os.path.dirname(__file__))
+
+    # Instead of connecting with Femtet, create
+    # a surrogate model. Read the CSV file created
+    # by the training data creation script to build
+    # the surrogate model.
+    fem = PoFBoTorchInterface(
+        history_path='training_data.csv'
+    )
+
+    # Set up the optimization object.
+    opt = OptunaOptimizer(
+        sampler_class=TPESampler,
+    )
+
+    # Set up the FEMOpt object.
+    femopt = FEMOpt(
+        fem=fem,
+        opt=opt,
+        history_path=f'optimized_result_target_{target}.csv'
+    )
+
+    # Set up the design variables.
+    # The upper and lower limits can differ from
+    # those in the training data creation script,
+    # but please note that extrapolation will
+    # occur outside the range that has not been
+    # trained, which may reduce the prediction
+    # accuracy of the surrogate model.
+    femopt.add_parameter('length', 0.1, 0.02, 0.2)
+    femopt.add_parameter('width', 0.01, 0.001, 0.02)
+
+    # If there are parameters that were set as
+    # design variables during training and wanted
+    # to fix during optimization, specify only the
+    # `initial_value` and set the `fix` argument True.
+    # You cannot add design variables that were not
+    # set during training for optimization.
+    femopt.add_parameter('base_radius', 0.008, fix=True)
+
+    # Specify the objective functions set during
+    # training that you want to optimize.
+    # You may provide the fun argument, but it will
+    # be overwritten during surrogate model creation,
+    # so it will be ignored.
+    # You cannot use objective functions that were
+    # not set during training for optimization.
+    obj_name = 'First Resonant Frequency (Hz)'
+    femopt.add_objective(
+        name=obj_name,
+        direction=target,
+    )
+
+    # Execute the optimization.
+    femopt.set_random_seed(42)
+    df = femopt.optimize(
+        n_trials=50,
+        confirm_before_exit=False
+    )
+
+    # Display the optimal solution.
+    prm_names = femopt.history.prm_names
+    obj_names = femopt.history.obj_names
+    prm_values = df[df['non_domi'] == True][prm_names].values[0]
+    obj_values = df[df['non_domi'] == True][obj_names].values[0]
+
+    message = f'''
+===== Optimization Results =====
+Target Value: {target}
+Prediction by Surrogate Model:
+'''
+    for name, value in zip(prm_names, prm_values):
+        message += f'  {name}: {value}\n'
+    for name, value in zip(obj_names, obj_values):
+        message += f'  {name}: {value}\n'
+
+    return message
+
+
+if __name__ == '__main__':
+    # Using the surrogate model created from the training data,
+    # we will find a design that results in a resonant frequency of 1000.
+    message_1000 = main(target=1000)
+
+    # Next, using the same surrogate model,
+    # we will find a design that results in a resonant frequency of 2000.
+    message_2000 = main(target=2000)
+
+    print(message_1000)
+    print(message_2000)

pyfemtet/opt/advanced_samples/surrogate_model/gal_ex13_optimize_with_surrogate_jp.py

@@ -0,0 +1,90 @@
+import os
+
+from optuna.samplers import TPESampler
+
+from pyfemtet.opt import FEMOpt, OptunaOptimizer
+from pyfemtet.opt.interface import PoFBoTorchInterface
+
+
+def main(target):
+
+    os.chdir(os.path.dirname(__file__))
+
+    # Femtet との接続の代わりに、サロゲートモデルを作成します。
+    # 学習データ作成スクリプトで作成した csv ファイルを読み込んで
+    # サロゲートモデルを作成します。
+    fem = PoFBoTorchInterface(
+        history_path='training_data.csv'
+    )
+
+    # 最適化用オブジェクトの設定を行います。
+    opt = OptunaOptimizer(
+        sampler_class=TPESampler,
+    )
+
+    # FEMOpt オブジェクトの設定を行います。
+    femopt = FEMOpt(
+        fem=fem,
+        opt=opt,
+        history_path=f'optimized_result_target_{target}.csv'
+    )
+
+    # 設計変数の設定を行います。
+    # 上下限は学習データ作成スクリプトと異なっても良いですが、
+    # 学習していない範囲は外挿となりサロゲートモデルによる
+    # 予測精度が低下することに注意してください。
+    femopt.add_parameter('length', 0.1, 0.02, 0.2)
+    femopt.add_parameter('width', 0.01, 0.001, 0.02)
+
+    # 学習時は設計変数としていたが最適化時に固定したいパラメータがある場合
+    # initial_value のみを指定して fix 引数を True にしてください。
+    # 学習時に設定しなかった設計変数を最適化時に追加することはできません。
+    femopt.add_parameter('base_radius', 0.008, fix=True)
+
+    # 学習時に設定した目的関数のうち
+    # 最適化したいものを指定します。
+    # fun 引数は与えてもいいですが、サロゲートモデル作成時に上書きされるため無視されます。
+    # 学習時に設定しなかった目的関数を最適化時に使用することはできません。
+    obj_name = '第一共振周波数(Hz)'
+    femopt.add_objective(
+        name=obj_name,
+        direction=target,
+    )
+
+    # 最適化を実行します。
+    femopt.set_random_seed(42)
+    df = femopt.optimize(
+        n_trials=50,
+        confirm_before_exit=False
+    )
+
+    # 最適解を表示します。
+    prm_names = femopt.history.prm_names
+    obj_names = femopt.history.obj_names
+    prm_values = df[df['non_domi'] == True][prm_names].values[0]
+    obj_values = df[df['non_domi'] == True][obj_names].values[0]
+
+    message = f'''
+===== 最適化結果 =====    
+ターゲット値: {target}
+サロゲートモデルによる予測:
+'''
+    for name, value in zip(prm_names, prm_values):
+        message += f'  {name}: {value}\n'
+    for name, value in zip(obj_names, obj_values):
+        message += f'  {name}: {value}\n'
+
+    return message
+
+
+if __name__ == '__main__':
+    # 学習データから作成したサロゲートモデルで
+    # 共振周波数が 1000 になる設計を見つけます。
+    message_1000 = main(target=1000)
+
+    # 続いて、同じサロゲートモデルで
+    # 共振周波数が 2000 になる設計を見つけます。
+    message_2000 = main(target=2000)
+
+    print(message_1000)
+    print(message_2000)

pyfemtet/opt/interface/__init__.py

@@ -3,6 +3,7 @@ from pyfemtet.opt.interface._base import NoFEM
 from pyfemtet.opt.interface._femtet import FemtetInterface
 from pyfemtet.opt.interface._femtet_with_sldworks import FemtetWithSolidworksInterface
 from pyfemtet.opt.interface._femtet_with_nx import FemtetWithNXInterface
+from pyfemtet.opt.interface._surrogate import PoFBoTorchInterface
 
 
 __all__ = [
@@ -11,4 +12,5 @@ __all__ = [
     'FemtetInterface',
     'FemtetWithNXInterface',
     'FemtetWithSolidworksInterface',
+    'PoFBoTorchInterface',
 ]

pyfemtet/opt/interface/_base.py

@@ -55,6 +55,9 @@ class FEMInterface(ABC):
     def load_objective(self, opt) -> None:  # opt: AbstractOptimizer
         raise NotImplementedError()
 
+    def load_constraint(self, opt) -> None:  # opt: AbstractOptimizer
+        raise NotImplementedError()
+
     def _setup_before_parallel(self, client) -> None:
         """Preprocessing before launching a dask worker (if implemented in concrete class).