logging for time_encoding_minute

datasig-ac-uk · Nov 14, 2023 · 3cedfbc · 3cedfbc
1 parent d87e40e
commit 3cedfbc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/nlpsig/data_preparation.py b/src/nlpsig/data_preparation.py
@@ -260,6 +260,10 @@ def _set_time_features(self) -> pd.DataFrame:
             # reset index so that we can return to the starting index afterwards
             self.df = self.df.reset_index()
 
+            # obtain time encoding for minute by computing the fraction of minute it is in
+            if self.verbose:
+                print("[INFO] Adding 'time_encoding_minute' feature...")
+
             # add a column for the time fraction of the minute
             self.df["time_encoding_minute"] = self.df["datetime"].map(
                 lambda t: self._time_fraction_minute(t)