Add method to create individuals

fau-advanced-separations · Feb 5, 2024 · 5ba63ba · 5ba63ba
1 parent a7229c8
commit 5ba63ba
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/CADETProcess/optimization/optimizationProblem.py b/CADETProcess/optimization/optimizationProblem.py
@@ -38,6 +38,7 @@
 
 from CADETProcess.metric import MetricBase
 
+from CADETProcess.optimization import Individual, Population
 from CADETProcess.optimization import ResultsCache
 
 
@@ -2851,6 +2852,23 @@ def create_initial_values(
 
         return np.array(values, ndmin=2)
 
+    @untransforms
+    @gets_dependent_values
+    def create_individual(self, x, f=None, g=None, m=None, cv=None, cv_tol=None):
+        x_indep = self.get_independent_values(x)
+        x_transformed = self.transform(x_indep)
+
+        ind = Individual(
+            x, f, g, m, cv, cv_tol, x_transformed,
+            self.independent_variable_names,
+            self.objective_labels,
+            self.nonlinear_constraint_labels,
+            self.meta_score_labels,
+            self.variable_names,
+        )
+
+        return ind
+
     @property
     def parameters(self):
         parameters = Dict()