Update stuff_model/feature_engineering.py

stuff_model/feature_engineering.py

@@ -120,12 +120,21 @@ def feature_engineering(df: pl.DataFrame) -> pl.DataFrame:
     
     # Calculate delta time (Δt)
     delta_t = (df["release_pos_y"] - df["y0"]) / df["vy0"]
-
+    # print((df["vx0"] * delta_t + 0.5 * df["ax"] * delta_t ** 2))
     # Corrected back-calculation of release_pos_x and release_pos_z
+
+    
+    df = df.with_columns(
+            pl.when(pl.col('pitcher_hand')== 'R')
+            .then(df["x0"] - df["vx0"] * delta_t - 0.5 * df["ax"] * delta_t ** 2)
+            .otherwise(df["x0"] + df["vx0"] * delta_t - 0.5 * df["ax"] * delta_t ** 2)
+            .alias('release_pos_x')
+        )
+
     df = df.with_columns([
-        (df["x0"] + df["vx0"] * delta_t + 0.5 * df["ax"] * delta_t ** 2).alias("release_pos_x"),
         (df["z0"] + df["vz0"] * delta_t + 0.5 * df["az"] * delta_t ** 2).alias("release_pos_z")
     ])
 
+
     
     return df