| { | |
| "num_players": 496, | |
| "hidden_dim": 64, | |
| "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_OneGraph", | |
| "framework": "PyTorch Geometric", | |
| "task": "binary_classification", | |
| "target": "goal", | |
| "node_types": [ | |
| "shooter", | |
| "goal", | |
| "goalkeeper" | |
| ], | |
| "edge_types": [ | |
| "('shooter', 'shoots_at', 'goal')", | |
| "('goal', 'rev_shoots_at', 'shooter')", | |
| "('shooter', 'faces', 'goalkeeper')", | |
| "('goalkeeper', 'rev_faces', 'shooter')" | |
| ], | |
| "graph_structure": "persistent_nodes_with_shot_indexed_edges" | |
| } |