dev

ML-KULeuven · Jun 4, 2024 · 97fc6f8 · 97fc6f8
1 parent 8aa96b1
commit 97fc6f8
Show file tree

Hide file tree

Showing 13 changed files with 1,309 additions and 791 deletions.
diff --git a/soccer_xg/__init__.py b/soccer_xg/__init__.py
@@ -4,7 +4,27 @@
 :copyright: (c) 2023 by DTAI KU Leuven.
 :license: Apache v2, see LICENSE for more details.
 """
-from soccer_xg.xg import XGModel
+from soccer_xg.data import Dataset, HDFDataset, SQLDataset
+from soccer_xg.xg import (
+    DatasetTransformer,
+    XGModel,
+    PenaltyXGModel,
+    FreekickXGModel,
+    BasicOpenplayXGModel,
+    AdvancedOpenplayXGModel,
+    StatsBombOpenplayXGModel,
+)
 
 __version__ = '0.0.1'
-__all__ = ['XGModel']
+__all__ = [
+    'Dataset',
+    'HDFDataset',
+    'SQLDataset',
+    'DatasetTransformer',
+    'XGModel',
+    'PenaltyXGModel',
+    'FreekickXGModel',
+    'BasicOpenplayXGModel',
+    'AdvancedOpenplayXGModel',
+    'StatsBombOpenplayXGModel',
+]