File size: 1,218 Bytes
3949a44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
{
"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"
} |