| { |
| "num_players": 496, |
| "hidden_dim": 32, |
| "dropout_rate": 0.3, |
| "num_heads": 4, |
| "num_layers": 3, |
| "use_norm": true, |
| "num_global_features": 18, |
| "global_feature_names": [ |
| "ball_closer_than_gk", |
| "body_part_name_Left Foot", |
| "body_part_name_Other", |
| "body_part_name_Right Foot", |
| "goal_dist_to_gk", |
| "minute", |
| "nearest_opponent_dist", |
| "nearest_teammate_dist", |
| "opponents_within_5m", |
| "play_pattern_name_From Counter", |
| "play_pattern_name_From Free Kick", |
| "play_pattern_name_From Goal Kick", |
| "play_pattern_name_From Keeper", |
| "play_pattern_name_From Kick Off", |
| "play_pattern_name_From Throw In", |
| "play_pattern_name_Other", |
| "play_pattern_name_Regular Play", |
| "teammates_within_5m" |
| ], |
| "model_type": "HeterogeneousGNN", |
| "framework": "PyTorch Geometric", |
| "task": "binary_classification", |
| "target": "goal", |
| "node_types": [ |
| "shooter", |
| "goal" |
| ], |
| "edge_types": [ |
| "('goal', 'distance', 'shooter')", |
| "('goal', 'angle_to_goal', 'shooter')", |
| "('goal', 'dist_to_gk', 'shooter')", |
| "('goal', 'angle_to_gk', 'shooter')" |
| ] |
| } |