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