@@ -31,25 +31,31 @@ class SearchAlgorithm:
31
31
32
32
def generate_fitness_functions (self , surrogate_models : list [Estimator ]) -> list [SearchFitnessFunction ]:
33
33
"""Generates the fitness function of the search space
34
- :param surrogate_models: """
35
- pass
34
+ :param surrogate_models: A list of CubicSplineRegressionEstimator generated for each edge of the DAG
35
+ :return: A list of fitness functions mapping to each of the surrogate models in the input"""
36
36
37
37
def search (self , fitness_functions : list [SearchFitnessFunction ], specification : CausalSpecification ) -> list :
38
- pass
38
+ """Function which implements a search routine which searches for the optimal fitness value for the specified
39
+ scenario
40
+ :param fitness_functions: The fitness function to be optimised
41
+ :param specification: The Causal Specification (combination of Scenario and Causal Dag)"""
39
42
40
43
41
44
class Simulator :
42
45
"""Class to be inherited with Simulator specific functions to start, shutdown and run the simulation with the give
43
46
config file"""
44
47
45
48
def startup (self , ** kwargs ):
46
- pass
49
+ """Function that when run, initialises and opens the Simulator"""
47
50
48
51
def shutdown (self , ** kwargs ):
49
- pass
52
+ """Function to safely exit and shutdown the Simulator"""
50
53
51
54
def run_with_config (self , configuration ) -> SimulationResult :
52
- pass
55
+ """Run the simulator with the given configuration and return the results in the structure of a
56
+ SimulationResult
57
+ :param configuration:
58
+ :return: Simulation results in the structure of the SimulationResult data class"""
53
59
54
60
55
61
class CausalSurrogateAssistedTestCase :
0 commit comments