prediction to docs

UnravelSports · Jul 25, 2024 · a3a17ca · a3a17ca
1 parent 044e253
commit a3a17ca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/1_kloppy_gnn_train.ipynb b/examples/1_kloppy_gnn_train.ipynb
@@ -123,6 +123,7 @@
     "\n",
     "| Parameter | Type | Description | Default |\n",
     "|-----------|------|-------------|---------|\n",
+    "| `prediction` | bool | When True use the converter to create Graph dataset to apply a pre-trained model to, no labels required. Defaults to False. | False |\n",
     "| `ball_carrier_threshold` | float | The distance threshold to determine the ball carrier in meters. If no ball carrier within ball_carrier_threshold, we skip the frame. | 25.0 |\n",
     "| `max_player_speed` | float | The maximum speed of a player in meters per second. Used for normalizing node features. | 12.0 |\n",
     "| `max_ball_speed` | float | The maximum speed of the ball in meters per second. Used for normalizing node features. | 28.0 |\n",