fix mfk mixed

Paul-Saves · Feb 29, 2024 · ac7034d · ac7034d
1 parent d2bc7cd
commit ac7034d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/smt/applications/mfk.py b/smt/applications/mfk.py
@@ -296,6 +296,9 @@ def _new_train_init(self):
             _, self.cat_features = compute_X_cont(
                 np.concatenate(xt, axis=0), self.design_space
             )
+            self.X_offset[self.cat_features] *=0
+            self.X_scale[self.cat_features] *=0
+            self.X_scale[self.cat_features] +=1
 
         nlevel = self.nlvl