XGBoost TPLF Model Training

import xgboost as xgb
import pandas as pd

# Prepare features: venue_code, injury_score, tplf_flag, firm_experience
X = pd.DataFrame(features)
y = df['severity_multiplier']

model = xgb.XGBRegressor(n_estimators=100, max_depth=5, learning_rate=0.05)
model.fit(X, y)
# Export model and evaluate prediction...