take all datasets

prepare.py

@@ -13,10 +13,13 @@ if __name__ == "__main__":
         print("Dataset loaded from cache.")
     else:
         # Load dataset using datasets.load_dataset()
-        ds = datasets.load_dataset("renumics/speech_commands_enriched", split="train")
+        ds_train = datasets.load_dataset("renumics/speech_commands_enriched", split="train")
+        ds_test = datasets.load_dataset("renumics/speech_commands_enriched", split="test")
+        ds_validation = datasets.load_dataset("renumics/speech_commands_enriched", split="validation")
+        joined_dataset_enrichment = datasets.concatenate_datasets([ds_train, ds_validation, ds_test])
         print("Dataset loaded using datasets.load_dataset().")
 
-        df = ds.to_pandas()
+        df = joined_dataset_enrichment.to_pandas()
 
         # Save dataset to cache
         with open(cache_file, "wb") as file: