diff --git a/code/experiments/utils/result_collection.py b/code/experiments/utils/result_collection.py
index 10626e5..9de054b 100644
--- a/code/experiments/utils/result_collection.py
+++ b/code/experiments/utils/result_collection.py
@@ -11,6 +11,7 @@ def __init__(
     ) -> None:
         self.data = data
         self.config = config
+        self.ds_type = config.ds_type
 
 
 class ResultCollection:
@@ -38,6 +39,7 @@ def save(self, t_file_prefix: str = "res"):
                 result = x.data
                 result["type_model"] = x.config.conf_model.type.name.lower()
                 result["transform"] = x.config.transforms.name.lower()
+                result["ds_type"] = x.config.ds_type.name.lower()
                 data.append(x.data)
 
             df = pd.DataFrame(data)