Spaces:
Configuration error
Configuration error
| """Comprehensive feature engineering from existing FormFav data.""" | |
| from __future__ import annotations | |
| import json | |
| from pathlib import Path | |
| from typing import Optional | |
| from trifecta_bro.data.models import RunnerModel, RaceModel | |
| def extract_features(runner: RunnerModel, race: RaceModel) -> dict: | |
| """Extract all possible features from runner data for ML model.""" | |
| features = {} | |
| # Basic features | |
| features['number'] = runner.number | |
| features['barrier'] = runner.barrier or 0 | |
| features['weight'] = runner.weight or 57.0 | |
| features['age'] = runner.age or 4 | |
| # Sex encoding | |
| sex = (runner.sex or '').lower() | |
| features['is_gelding'] = 1 if 'gelding' in sex else 0 | |
| features['is_mare'] = 1 if 'mare' in sex else 0 | |
| features['is_filly'] = 1 if 'filly' in sex else 0 | |
| features['is_colt'] = 1 if 'colt' in sex else 0 | |
| # Career stats | |
| overall = runner.stats.get('overall', {}) | |
| features['career_starts'] = overall.get('starts', 0) or 0 | |
| features['career_wins'] = overall.get('wins', 0) or 0 | |
| features['career_places'] = overall.get('places', 0) or 0 | |
| features['career_win_pct'] = overall.get('winPercent', 0) or 0 | |
| features['career_place_pct'] = overall.get('placePercent', 0) or 0 | |
| # Track stats | |
| track = runner.stats.get('track', {}) | |
| features['track_starts'] = track.get('starts', 0) or 0 | |
| features['track_wins'] = track.get('wins', 0) or 0 | |
| features['track_win_pct'] = track.get('winPercent', 0) or 0 | |
| features['track_place_pct'] = track.get('placePercent', 0) or 0 | |
| # Distance stats | |
| distance = runner.stats.get('distance', {}) | |
| features['distance_starts'] = distance.get('starts', 0) or 0 | |
| features['distance_wins'] = distance.get('wins', 0) or 0 | |
| features['distance_win_pct'] = distance.get('winPercent', 0) or 0 | |
| features['distance_place_pct'] = distance.get('placePercent', 0) or 0 | |
| # Track + Distance combo | |
| track_dist = runner.stats.get('trackDistance', {}) | |
| features['track_distance_starts'] = track_dist.get('starts', 0) or 0 | |
| features['track_distance_wins'] = track_dist.get('wins', 0) or 0 | |
| features['track_distance_win_pct'] = track_dist.get('winPercent', 0) or 0 | |
| features['track_distance_place_pct'] = track_dist.get('placePercent', 0) or 0 | |
| # Condition stats (match by race condition) | |
| cond = (race.condition or '').lower() | |
| cond_stats = runner.stats.get('conditions', {}) | |
| cond_key = None | |
| for key in cond_stats: | |
| if cond and cond[:4] in key.lower(): | |
| cond_key = key | |
| break | |
| if cond_key: | |
| cs = cond_stats[cond_key] | |
| features['cond_starts'] = cs.get('starts', 0) or 0 | |
| features['cond_wins'] = cs.get('wins', 0) or 0 | |
| features['cond_win_pct'] = cs.get('winPercent', 0) or 0 | |
| features['cond_place_pct'] = cs.get('placePercent', 0) or 0 | |
| else: | |
| features['cond_starts'] = 0 | |
| features['cond_wins'] = 0 | |
| features['cond_win_pct'] = 0 | |
| features['cond_place_pct'] = 0 | |
| # First-up / Second-up stats | |
| first_up = runner.stats.get('firstUp', {}) | |
| second_up = runner.stats.get('secondUp', {}) | |
| features['first_up_starts'] = first_up.get('starts', 0) or 0 | |
| features['first_up_wins'] = first_up.get('wins', 0) or 0 | |
| features['first_up_win_pct'] = first_up.get('winPercent', 0) or 0 | |
| features['second_up_starts'] = second_up.get('starts', 0) or 0 | |
| features['second_up_wins'] = second_up.get('wins', 0) or 0 | |
| features['second_up_win_pct'] = second_up.get('winPercent', 0) or 0 | |
| # Form analysis | |
| form = runner.form or '' | |
| features['form_length'] = len(form) | |
| # Recent form (last 5) | |
| recent = form[-5:] if len(form) >= 5 else form | |
| features['recent_wins'] = sum(1 for c in recent if c == '1') | |
| features['recent_places'] = sum(1 for c in recent if c in ['1', '2', '3']) | |
| features['recent_win_pct'] = features['recent_wins'] / len(recent) if recent else 0 | |
| features['recent_place_pct'] = features['recent_places'] / len(recent) if recent else 0 | |
| # Very recent (last 3) | |
| very_recent = form[-3:] if len(form) >= 3 else form | |
| features['very_recent_wins'] = sum(1 for c in very_recent if c == '1') | |
| features['very_recent_win_pct'] = features['very_recent_wins'] / len(very_recent) if very_recent else 0 | |
| # Last start result | |
| features['last_start_pos'] = _parse_last_position(form[-1]) if form else 0 | |
| # Career prize money | |
| prize = runner.career_prize_money | |
| if prize: | |
| try: | |
| features['career_prize'] = float(str(prize).replace('$', '').replace(',', '')) | |
| except: | |
| features['career_prize'] = 0 | |
| else: | |
| features['career_prize'] = 0 | |
| # Prize money per start | |
| features['prize_per_start'] = features['career_prize'] / features['career_starts'] if features['career_starts'] > 0 else 0 | |
| # Race features | |
| features['num_runners'] = race.number_of_runners or 10 | |
| # Distance in metres | |
| distance_str = race.distance or '1200m' | |
| try: | |
| features['distance_metres'] = int(''.join(filter(str.isdigit, str(distance_str)))) | |
| except: | |
| features['distance_metres'] = 1200 | |
| # Barrier advantage (relative to field) | |
| features['barrier_central'] = abs(features['barrier'] - features['num_runners'] / 2) | |
| # Weight relative to field | |
| features['weight_relative'] = features['weight'] - 57.0 # 57kg is average | |
| return features | |
| def _parse_last_position(char: str) -> int: | |
| """Parse a form character to a finishing position.""" | |
| if char == '1': | |
| return 1 | |
| elif char == '2': | |
| return 2 | |
| elif char == '3': | |
| return 3 | |
| elif char == '4': | |
| return 4 | |
| elif char == '5': | |
| return 5 | |
| elif char == '6': | |
| return 6 | |
| elif char == '7': | |
| return 7 | |
| elif char == '8': | |
| return 8 | |
| elif char == '9': | |
| return 9 | |
| elif char == '0': | |
| return 10 # spell | |
| elif char in ['x', 'X']: | |
| return 0 # spell | |
| elif char in ['f', 'F']: | |
| return 0 # fell | |
| elif char in ['p', 'P']: | |
| return 0 # pulled up | |
| else: | |
| return 0 | |