MultiObjectiveSurrogateOptimization/interface/_ca1_culture.py at main · GazzolaLab/MultiObjectiveSurrogateOptimization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import os

from machinable import Interface, get


class _Ca1Culture(Interface):
    def launch(self):
        baseline = [
            """~from_protocol(
                './benchmarks/hippocampal_dissociated_culture/config/optimize_network.yaml',
                302,
                mechanisms_path="benchmarks/hippocampal_dissociated_culture/mechanisms",
                spike_input_path="Slice/CA1_Slice_100_dynamical_response_features_20250912.h5",
                spike_input_namespaces=["Spatiotemporal Feature Spikes drc_features_20250912"],
                spike_input_attr='Spike Train',
                dataset_prefix=".",
                config_prefix="./benchmarks/hippocampal_dissociated_culture/config",
                coordinates_namespace="Generated Coordinates",
                max_walltime_hours=48,
                io_size=1,
                )""",
            {
                "dopt_params": {
                    "optimizer_name": "nsga2",
                    "initial_method": "mc",
                    "n_initial": 1,
                    "initial_maxiter": 10,
                    "n_epochs": 50,
                    "population_size": 100,
                    "num_generations": 10,
                    "termination_conditions": True,
                    "resample_fraction": 0.1,
                    "feasibility_method_name": None,
                    "feasibility_method_kwargs": {},
                    "save": True,
                    "save_surrogate_evals": True,
                }
            },
        ]

        # gpr
        get(
            "interface.sopt_culture",
            baseline
            + [
                {
                    "dopt_params.surrogate_method_name": "gpr",
                    "dopt_params.surrogate_method_kwargs": {},
                },
            ],
        ).launch()

        # megp
        get(
            "interface.sopt_culture",
            baseline
            + [
                {
                    "dopt_params.surrogate_method_name": "megp",
                    "dopt_params.surrogate_method_kwargs": {},
                },
            ],
        ).launch()

        # joint
        get(
            "interface.sopt_culture",
            baseline
            + [
                {
                    "dopt_params": {
                        "surrogate_custom_training": "models.ops.joint",
                        "surrogate_custom_training_kwargs": {
                            "mode": "c+o",
                            "sensitivity": True,
                            "backbone": "fttransformer",
                            "queue": "/scratch1/08818/fg14/MultiObjectiveSurrogateOptimization/queue",
                        },
                    }
                },
            ],
        ).launch()