feat: Optimize LSTM with Attention, add Stacking Ensemble and SHAP analysis

2026-02-10 18:59:48 +08:00
parent a355239861
commit 706940d8d3
12 changed files with 1087 additions and 0 deletions
--- a/src/training/train.py
+++ b/src/training/train.py
@@ -93,6 +93,9 @@ def preprocess_features(df):
    df['is_t'] = (df['team_num'] == 2).astype(int)
    df['is_ct'] = (df['team_num'] == 3).astype(int)
    
+    # Fill NA in 'is_alive' before conversion
+    df['is_alive'] = df['is_alive'].fillna(0)
+
    # Calculate player specific metrics
    df['t_alive'] = df['is_t'] * df['is_alive'].astype(int)
    df['ct_alive'] = df['is_ct'] * df['is_alive'].astype(int)