Fix warning

sdv-dev · Apr 25, 2024 · 63ffd73 · 63ffd73
1 parent 825a643
commit 63ffd73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ctgan/data_transformer.py b/ctgan/data_transformer.py
@@ -189,7 +189,7 @@ def transform(self, raw_data):
 
     def _inverse_transform_continuous(self, column_transform_info, column_data, sigmas, st):
         gm = column_transform_info.transform
-        data = pd.DataFrame(column_data[:, :2], columns=list(gm.get_output_sdtypes()))
+        data = pd.DataFrame(column_data[:, :2], columns=list(gm.get_output_sdtypes())).astype(float)
         data[data.columns[1]] = np.argmax(column_data[:, 1:], axis=1)
         if sigmas is not None:
             selected_normalized_value = np.random.normal(data.iloc[:, 0], sigmas[st])