Merge branch 'feature/discretizer' into 'main'

feature/disretizer See merge request ai-lab-pmo/mltools/recsys/RePlay!236
sb-ai-lab · Nov 13, 2024 · 5bb07f9 · 5bb07f9
2 parents 4e95eea + 85f3959
commit 5bb07f9
Show file tree

Hide file tree

Showing 4 changed files with 1,094 additions and 0 deletions.
diff --git a/replay/preprocessing/__init__.py b/replay/preprocessing/__init__.py
@@ -6,6 +6,12 @@
 """
 
 from .converter import CSRConverter
+from .discretizer import (
+    Discretizer,
+    GreedyDiscretizingRule,
+    HandleInvalidStrategies,
+    QuantileDiscretizingRule,
+)
 from .history_based_fp import (
     ConditionalPopularityProcessor,
     EmptyFeatureProcessor,