print first row of data files

Author: aditya0by0

chebai/preprocessing/datasets/base.py

@@ -474,6 +474,12 @@ def _set_processed_data_props(self):
         )
         data_pt = torch.load(pt_file_path, weights_only=False)
 
+        print(
+            "First row of transformed data (.pt pytorch file) :\n",
+            data_pt[0],
+            end="\n\n",
+        )
+
         self._num_of_labels = len(data_pt[0]["labels"])
         self._feature_vector_size = max(len(d["features"]) for d in data_pt)
 
@@ -936,6 +942,9 @@ def _get_data_size(input_file_path: str) -> int:
         with open(input_file_path, "rb") as f:
             df = pd.read_pickle(f)
             print(f"Processed data size ({input_file_path}): {len(df)} rows")
+            print(
+                "First row of processed data (pickle file):\n", df.iloc[0], end="\n\n"
+            )
             return len(df)
 
     @abstractmethod