gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-18 21:38:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(Data): added day/month/dayofweek features

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2021-11-14 17:24:56 +01:00
parent 9d342d6039
commit 478a15a7cb
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
load_data.py
+4
View File
@@ -31,6 +31,7 @@ def __load_df(path: str, prefix: str, add_features: bool, log_returns: bool) ->
df['vol_10'] = df['returns'].rolling(10).std()*(252**0.5) df['vol_10'] = df['returns'].rolling(10).std()*(252**0.5)
df['vol_20'] = df['returns'].rolling(20).std()*(252**0.5) df['vol_20'] = df['returns'].rolling(20).std()*(252**0.5)
df['vol_30'] = df['returns'].rolling(30).std()*(252**0.5) df['vol_30'] = df['returns'].rolling(30).std()*(252**0.5)
df['vol_60'] = df['returns'].rolling(30).std()*(252**0.5)
# momentum (10, 20, 30, 60, 90 days) # momentum (10, 20, 30, 60, 90 days)
if log_returns: if log_returns:
@@ -51,3 +52,6 @@ def __load_df(path: str, prefix: str, add_features: bool, log_returns: bool) ->
return df return df
# %% # %%
def create_target_cum_forward_returns(df: pd.DataFrame, source_column: str, period: int) -> pd.DataFrame:
df['target'] = df[source_column].diff(period).shift(-period)
return df