chore: refactor to better map end to end

This commit is contained in:
2026-01-12 10:09:55 +01:00
parent f9bf3de71e
commit 8b429b7a8e
5 changed files with 153 additions and 61 deletions

View File

@@ -135,6 +135,7 @@ class ExtractSessionFeaturesStep(BaseContextStep):
Vectorized session feature extraction - replaces O(n^2) per-row loop.
Input: interactions_df
Output: session-level feature matrix
THIS is our main mapping from tau (trajectory) to some features vector theta - we need to do this very well. This is what will go into demand esimation.
"""
def transform(self, X: pd.DataFrame) -> pd.DataFrame: