| # tgn-sym-dynamic |
|
|
| JAX checkpoint for the `TgnSymDynamic` football pass prediction model. |
|
|
| ## Checkpoint |
|
|
| - Model class: `TgnSymDynamic` |
| - Framework: `jax` |
| - Run timestamp: `20260430_084243` |
| - Parameters: `model_params.joblib` |
| - Hyperparameters: `config.json` |
| - Metrics: `metrics.json` |
|
|
| ## Validation Metrics |
|
|
| ```json |
| { |
| "val/loss": 0.928921639919281, |
| "val/intention_top1_accuracy": 0.7339416146278381, |
| "val/intention_top2_accuracy": 0.8846715688705444, |
| "val/intention_top3_accuracy": 0.9397810697555542, |
| "val/intention_mrr": 0.8394966125488281, |
| "val/outcome_top1_accuracy": 0.6613138914108276, |
| "val/outcome_top2_accuracy": 0.8071168065071106, |
| "val/outcome_top3_accuracy": 0.8708029389381409, |
| "val/outcome_mrr": 0.777962327003479 |
| } |
| ``` |
|
|
| ## Loading |
|
|
| ```python |
| import json |
| from pathlib import Path |
| |
| import joblib |
| |
| checkpoint_path = Path("path/to/checkpoint") |
| |
| with open(checkpoint_path / "config.json") as f: |
| config = json.load(f) |
| |
| params = joblib.load(checkpoint_path / "model_params.joblib") |
| ``` |
|
|
