Dev fix syn data bug (#548)

alibaba · Mar 18, 2023 · 0bf3c11 · 0bf3c11
1 parent c6a7de4
commit 0bf3c11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/federatedscope/vertical_fl/dataloader/dataloader.py b/federatedscope/vertical_fl/dataloader/dataloader.py
@@ -48,7 +48,7 @@ def load_vertical_data(config=None, generate=False):
         INSTANCE_NUM = 1000
         TRAIN_SPLIT = 0.9
 
-        total_dims = np.sum(config.vertical.dims)
+        total_dims = config.vertical.dims[-1]
         theta = np.random.uniform(low=-1.0, high=1.0, size=(total_dims, 1))
         x = np.random.choice([-1.0, 1.0, -2.0, 2.0, -3.0, 3.0],
                              size=(INSTANCE_NUM, total_dims))