fix fatal bugs

caradryanl · May 17, 2024 · 9f82171 · 9f82171
1 parent 9f6a3b0
commit 9f82171
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/diffusers/scripts/train_gsa.py b/diffusers/scripts/train_gsa.py
@@ -81,6 +81,7 @@ def preprocess(member, non_member):
     x = np.vstack((member, non_member))
     y = np.concatenate((member_y_np, nonmember_y_np))
     x = preprocessing.scale(x)
+    x = np.nan_to_num(x, nan=0)
     return x, y
 
 def train_xgboost(member_features, nonmember_features):