Fix LHS on TSEMO (#129)

sustainable-processes · Apr 21, 2021 · fa785af · fa785af
1 parent 6b846db
commit fa785af
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/summit/strategies/tsemo.py b/summit/strategies/tsemo.py
@@ -163,7 +163,7 @@ def suggest_experiments(self, num_experiments, prev_res: DataSet = None, **kwarg
 
         # Suggest lhs initial design or append new experiments to previous experiments
         if prev_res is None:
-            lhs = LHS(self.domain)
+            lhs = LHS(self.domain, categorical_method="descriptors")
             self.iterations += 1
             k = num_experiments if num_experiments > 1 else 2
             return lhs.suggest_experiments(k, criterion="maximin")
@@ -173,9 +173,8 @@ def suggest_experiments(self, num_experiments, prev_res: DataSet = None, **kwarg
             self.all_experiments = self.all_experiments.append(prev_res)
 
         if self.all_experiments.shape[0] <= 3:
-            lhs = LHS(self.domain)
+            lhs = LHS(self.domain, categorical_method="descriptors")
             self.iterations += 1
-            self.all_experiments = prev_res
             return lhs.suggest_experiments(num_experiments)
 
         # Get inputs (decision variables) and outputs (objectives)