2022-02-17 19:22:17 +01:00
|
|
|
from .labellers.fixed_time_three_class_balanced import (
|
|
|
|
|
FixedTimeHorionThreeClassBalancedEventLabeller,
|
|
|
|
|
)
|
|
|
|
|
from .labellers.fixed_time_three_class_imbalanced import (
|
|
|
|
|
FixedTimeHorionThreeClassImbalancedEventLabeller,
|
|
|
|
|
)
|
2022-01-26 23:22:43 +01:00
|
|
|
from .labellers.fixed_time_two_class import FixedTimeHorionTwoClassEventLabeller
|
|
|
|
|
|
|
|
|
|
labellers_map = dict(
|
2022-02-17 19:22:17 +01:00
|
|
|
two_class=FixedTimeHorionTwoClassEventLabeller,
|
|
|
|
|
three_class_balanced=FixedTimeHorionThreeClassBalancedEventLabeller,
|
|
|
|
|
three_class_imbalanced=FixedTimeHorionThreeClassImbalancedEventLabeller,
|
|
|
|
|
)
|
