Files
drift/labeling/labellers_map.py
T
2022-01-26 23:22:43 +01:00

9 lines
526 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()
)