From ac7034d03ab72efb0769650ada01a5f3eb800f57 Mon Sep 17 00:00:00 2001 From: Paul-Saves Date: Thu, 29 Feb 2024 18:37:26 +0100 Subject: [PATCH] fix mfk mixed --- smt/applications/mfk.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/smt/applications/mfk.py b/smt/applications/mfk.py index 604e9ae7a..58131143e 100644 --- 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