Update data_transformer.py

hitsz-ids · Nov 26, 2024 · 6551756 · 6551756
1 parent 6b51f28
commit 6551756
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdgx/models/components/optimize/sdv_ctgan/data_transformer.py b/sdgx/models/components/optimize/sdv_ctgan/data_transformer.py
@@ -66,7 +66,7 @@ def _fit_continuous(self, data):
             column_name=column_name,
             column_type="continuous",
             transform=gm,
-            output_info=[SpanInfo(1, "tanh"), SpanInfo(num_components, "softmax")],  # 贝叶斯gmm，多个正态分布，选择一个
+            output_info=[SpanInfo(1, "tanh"), SpanInfo(num_components, "softmax")], 
             output_dimensions=1 + num_components,
         )