AlgoX commited on
Commit
84953b4
·
1 Parent(s): c9926c2

feat : add function -> convert dataframe to tensor

Browse files
Files changed (1) hide show
  1. data_prep/data_load.py +7 -0
data_prep/data_load.py CHANGED
@@ -1,4 +1,5 @@
1
  import pandas as pd
 
2
 
3
  from data_prep.basic_feat import add_daily_return , indexify_date
4
  from data_prep.features import add_all_indicators
@@ -12,3 +13,9 @@ def prepare_data(path : str) -> pd.DataFrame:
12
  return add_all_indicators(df)
13
 
14
 
 
 
 
 
 
 
 
1
  import pandas as pd
2
+ import torch
3
 
4
  from data_prep.basic_feat import add_daily_return , indexify_date
5
  from data_prep.features import add_all_indicators
 
13
  return add_all_indicators(df)
14
 
15
 
16
+ def dataframe_to_tensor(df : pd.DataFrame) -> torch.Tensor:
17
+ x_np = df.to_numpy()
18
+ return torch.from_numpy(x_np)
19
+
20
+
21
+