Skip to content

Commit

Permalink
Update preprocess.py
Browse files Browse the repository at this point in the history
  • Loading branch information
wukan1986 committed Mar 12, 2024
1 parent 453238c commit fa77fbb
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions polars_ta/wq/preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,3 +110,8 @@ def cs_neutralize_residual_multiple(y: Expr, *more_x: Expr) -> Expr:
正则列需要通过`pl.struct`传输,比之前整体转`pl.struct`能支持复杂公式
"""
return map_batches([y, *more_x], lambda xx: residual_multiple(xx, False))


def cs_neutralize_residual(y: Expr, *more_x: Expr) -> Expr:
"""回归"""
return cs_neutralize_residual_multiple(y, *more_x)

0 comments on commit fa77fbb

Please sign in to comment.