Update models/pitch_sequence_model.py

models/pitch_sequence_model.py

@@ -19,12 +19,43 @@ def build_sequence_features(
         except Exception:
             return default
 
+    def _safe_float(value: Any, default: float | None = None) -> float | None:
+        try:
+            if value is None:
+                return default
+            text = str(value).strip().lower()
+            if text in {"", "nan", "none"}:
+                return default
+            return float(value)
+        except Exception:
+            return default
+
     balls = _safe_int(game_row.get("balls"), 0)
     strikes = _safe_int(game_row.get("strikes"), 0)
     outs = _safe_int(game_row.get("outs"), 0)
 
     last_pitch_type = str(game_row.get("pitch_type", "") or "").strip().lower()
 
+    avg_release_speed = _safe_float(pitcher_row.get("avg_release_speed"))
+    avg_release_spin_rate = _safe_float(pitcher_row.get("avg_release_spin_rate"))
+    avg_release_extension = _safe_float(pitcher_row.get("avg_release_extension"))
+
+    last_pitch_velocity = _safe_float(game_row.get("pitch_velocity"))
+    last_pitch_spin_rate = _safe_float(game_row.get("pitch_spin_rate"))
+    last_pitch_extension = _safe_float(game_row.get("pitch_extension"))
+
+    velo_delta_from_baseline = None
+    if last_pitch_velocity is not None and avg_release_speed is not None:
+        velo_delta_from_baseline = last_pitch_velocity - avg_release_speed
+
+    spin_delta_from_baseline = None
+    if last_pitch_spin_rate is not None and avg_release_spin_rate is not None:
+        spin_delta_from_baseline = last_pitch_spin_rate - avg_release_spin_rate
+
+    extension_delta_from_baseline = None
+    if last_pitch_extension is not None and avg_release_extension is not None:
+        extension_delta_from_baseline = last_pitch_extension - avg_release_extension
+
     return {
         "balls": balls,
         "strikes": strikes,
@@ -32,12 +63,20 @@ def build_sequence_features(
         "runner_on_1b": bool(game_row.get("runner_on_1b", False)),
         "runner_on_2b": bool(game_row.get("runner_on_2b", False)),
         "runner_on_3b": bool(game_row.get("runner_on_3b", False)),
-        "pitcher_avg_release_speed": pitcher_row.get("avg_release_speed"),
+        "pitcher_avg_release_speed": avg_release_speed,
+        "pitcher_avg_release_spin_rate": avg_release_spin_rate,
+        "pitcher_avg_release_extension": avg_release_extension,
         "pitcher_ev_allowed": pitcher_row.get("ev_allowed"),
         "batter_ev90": batter_row.get("ev90"),
         "batter_barrel_rate": batter_row.get("barrel_rate"),
         "batter_hard_hit_rate": batter_row.get("hard_hit_rate"),
         "last_pitch_type": last_pitch_type,
+        "last_pitch_velocity": last_pitch_velocity,
+        "last_pitch_spin_rate": last_pitch_spin_rate,
+        "last_pitch_extension": last_pitch_extension,
+        "velo_delta_from_baseline": velo_delta_from_baseline,
+        "spin_delta_from_baseline": spin_delta_from_baseline,
+        "extension_delta_from_baseline": extension_delta_from_baseline,
     }