2022-01-26 23:22:43 +01:00
|
|
|
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(
|
2022-02-17 16:36:35 +01:00
|
|
|
two_class = FixedTimeHorionTwoClassEventLabeller,
|
|
|
|
|
three_class_balanced = FixedTimeHorionThreeClassBalancedEventLabeller,
|
|
|
|
|
three_class_imbalanced = FixedTimeHorionThreeClassImbalancedEventLabeller
|
2022-01-26 23:22:43 +01:00
|
|
|
)
|