Skip to content

Commit 823c60a

Browse files
author
miranov25
committed
feat: create groupby_regression package structure
- Add package __init__.py with version info - Create tests/ subdirectory with __init__.py - Create benchmarks/ subdirectory with __init__.py - Create docs/ and benchmark_results/ directories - Prepare for file migration with git mv Imports in __init__.py commented out until files are moved.
1 parent 718ed1d commit 823c60a

File tree

3 files changed

+42
-0
lines changed

3 files changed

+42
-0
lines changed
Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
"""
2+
GroupBy Regression Package
3+
4+
Provides two implementations:
5+
- Robust (groupby_regression.py): Production-proven, full features, custom fitters
6+
- Optimized (groupby_regression_optimized.py): Speed-optimized (v2/v3/v4)
7+
8+
Quick Start:
9+
# Robust implementation (battle-tested)
10+
from dfextensions.groupby_regression import GroupByRegressor
11+
_, dfGB = GroupByRegressor.make_parallel_fit(...)
12+
13+
# Fast implementation (17-200× faster)
14+
from dfextensions.groupby_regression import make_parallel_fit_v4
15+
_, dfGB = make_parallel_fit_v4(...)
16+
17+
See docs/README.md for choosing between implementations.
18+
"""
19+
20+
# Import main classes from modules (will add after files are moved)
21+
# from .groupby_regression import GroupByRegressor
22+
# from .groupby_regression_optimized import (
23+
# make_parallel_fit_v2,
24+
# make_parallel_fit_v3,
25+
# make_parallel_fit_v4,
26+
# GroupByRegressorOptimized,
27+
# )
28+
29+
# Version info
30+
__version__ = '2.0.0'
31+
__author__ = 'Marian Ivanov'
32+
33+
# Expose at package level (will uncomment after files are moved)
34+
# __all__ = [
35+
# 'GroupByRegressor',
36+
# 'make_parallel_fit_v2',
37+
# 'make_parallel_fit_v3',
38+
# 'make_parallel_fit_v4',
39+
# 'GroupByRegressorOptimized',
40+
# ]
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
"""Benchmark suite for groupby_regression package"""
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
"""Test suite for groupby_regression package"""

0 commit comments

Comments
 (0)