Merge pull request #54 from compneurobilbao/hotfix-toy-data-precision

ENH: Changed synthetic data and generation to 3 floating point decimals
compneurobilbao · Oct 1, 2024 · fece052 · fece052
2 parents 48b3b87 + 5e74522
commit fece052
Show file tree

Hide file tree

Showing 4 changed files with 3,004 additions and 3,004 deletions.
diff --git a/src/ageml/datasets/synthetic_data.py b/src/ageml/datasets/synthetic_data.py
@@ -217,10 +217,10 @@ def generate_synthetic_data(save: bool = False, output_dir: str = None):
         covar_data_path = os.path.join(save_path, "toy_covar.csv")
         systems_path = os.path.join(save_path, "toy_systems.txt")
 
-        all_synthetic_data.to_csv(features_path, index=True, header=True)
-        clinical_data.to_csv(clinical_path, index=True, header=True)
-        factors_data.to_csv(factors_path, index=True, header=True)
-        covar_data.to_csv(covar_data_path, index=True, header=True)
+        all_synthetic_data.to_csv(features_path, index=True, header=True, float_format="%.3f")
+        clinical_data.to_csv(clinical_path, index=True, header=True, float_format="%.3f")
+        factors_data.to_csv(factors_path, index=True, header=True, float_format="%.3f")
+        covar_data.to_csv(covar_data_path, index=True, header=True, float_format="%.3f")
 
         # Write systems data to a .txt file exactly as it is
         with open(systems_path, "w") as f: