gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-07-27 18:57:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(Labeling): use X.loc instead of X.filter, remove Nan from event_start_times

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2022-03-12 13:27:44 +01:00
parent b656f790f5
commit c1d79ac8c2
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
labeling/labellers/fixed_time_three_class_balanced.py
+1 -1
View File
@@ -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):
labeling/labellers/fixed_time_three_class_imbalanced.py
+1 -1
View File
@@ -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):
labeling/labellers/fixed_time_two_class.py
+1 -1
View File
@@ -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:
labeling/process.py
+1 -1
View File
@@ -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"]