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()
)