Files
drift/labeling/labellers_map.py
T
Mark Aron Szulyovszky 8dd2d88740 chore(Linter): reformatted code with black (#211)
* chore(Linter): reformatted code with black

* Create black.yaml
2022-02-17 19:22:17 +01:00

14 lines
534 B
Python

from .labellers.fixed_time_three_class_balanced import (
FixedTimeHorionThreeClassBalancedEventLabeller,
)
from .labellers.fixed_time_three_class_imbalanced import (
FixedTimeHorionThreeClassImbalancedEventLabeller,
)
from .labellers.fixed_time_two_class import FixedTimeHorionTwoClassEventLabeller
labellers_map = dict(
two_class=FixedTimeHorionTwoClassEventLabeller,
three_class_balanced=FixedTimeHorionThreeClassBalancedEventLabeller,
three_class_imbalanced=FixedTimeHorionThreeClassImbalancedEventLabeller,
)