created correct_sign function

AC-BO-Hackathon · Mar 27, 2024 · 347076a · 347076a
1 parent 4c6dbaf
commit 347076a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/mobo_qm9.py b/src/mobo_qm9.py
@@ -104,7 +104,8 @@ def get_surrogate_model(self, X, y, kernel_type='RBF'):
 
         return model
 
-        y_copy = y.copy()
+    def correct_sign(self,Y)
+        y_copy = Y.copy()
         for idx, mask in enumerate(self.params.target_bools):
             if not mask:
                 y_copy[:, idx] *= -1