-
-
Notifications
You must be signed in to change notification settings - Fork 179
/
__init__.py
36 lines (31 loc) 路 1016 Bytes
/
__init__.py
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
# -*- coding: utf-8 -*-
"""Result trackers in PyKEEN."""
from class_resolver import Resolver, get_subclasses
from .base import ConsoleResultTracker, ResultTracker
from .file import CSVResultTracker, FileResultTracker, JSONResultTracker
from .mlflow import MLFlowResultTracker
from .neptune import NeptuneResultTracker
from .tensorboard import TensorBoardResultTracker
from .wandb import WANDBResultTracker
__all__ = [
# Base classes
'ResultTracker',
'FileResultTracker',
# Concrete classes
'MLFlowResultTracker',
'NeptuneResultTracker',
'WANDBResultTracker',
'JSONResultTracker',
'CSVResultTracker',
'TensorBoardResultTracker',
'ConsoleResultTracker',
# Utilities
'tracker_resolver',
]
_RESULT_TRACKER_SUFFIX = 'ResultTracker'
_TRACKERS = [
tracker
for tracker in get_subclasses(ResultTracker)
if tracker not in {FileResultTracker}
]
tracker_resolver = Resolver(_TRACKERS, base=ResultTracker, default=ResultTracker, suffix=_RESULT_TRACKER_SUFFIX)