diff --git a/labeling/labellers/fixed_time_three_class_balanced.py b/labeling/labellers/fixed_time_three_class_balanced.py index bece946..07ae563 100644 --- a/labeling/labellers/fixed_time_three_class_balanced.py +++ b/labeling/labellers/fixed_time_three_class_balanced.py @@ -18,7 +18,7 @@ class FixedTimeHorionThreeClassBalancedEventLabeller(EventLabeller): forward_returns = create_forward_returns(returns, self.time_horizon) cutoff_point = returns.index[-self.time_horizon] - event_start_times[event_start_times < cutoff_point] + event_start_times = event_start_times[event_start_times < cutoff_point] event_candidates = forward_returns[event_start_times] def get_bins_threeway(x): diff --git a/labeling/labellers/fixed_time_three_class_imbalanced.py b/labeling/labellers/fixed_time_three_class_imbalanced.py index b2278e6..d9aa107 100644 --- a/labeling/labellers/fixed_time_three_class_imbalanced.py +++ b/labeling/labellers/fixed_time_three_class_imbalanced.py @@ -17,7 +17,7 @@ class FixedTimeHorionThreeClassImbalancedEventLabeller(EventLabeller): forward_returns = create_forward_returns(returns, self.time_horizon) cutoff_point = returns.index[-self.time_horizon] - event_start_times[event_start_times < cutoff_point] + event_start_times = event_start_times[event_start_times < cutoff_point] event_candidates = forward_returns[event_start_times] def get_bins_threeway(x): diff --git a/labeling/labellers/fixed_time_two_class.py b/labeling/labellers/fixed_time_two_class.py index ca9cb17..aee5d52 100644 --- a/labeling/labellers/fixed_time_two_class.py +++ b/labeling/labellers/fixed_time_two_class.py @@ -18,7 +18,7 @@ class FixedTimeHorionTwoClassEventLabeller(EventLabeller): forward_returns = create_forward_returns(returns, self.time_horizon) cutoff_point = returns.index[-self.time_horizon] - event_start_times[event_start_times < cutoff_point] + event_start_times = event_start_times[event_start_times < cutoff_point] event_candidates = forward_returns[event_start_times] def get_class_binary(x: float) -> int: diff --git a/labeling/process.py b/labeling/process.py index d979b25..5f573e4 100644 --- a/labeling/process.py +++ b/labeling/process.py @@ -27,7 +27,7 @@ def label_data( "% of overlapping events", ) - X = X.filter(items=events.index, axis=0) + X = X.loc[events.index] y = events["label"] forward_returns = events["returns"]