progress

src/tools/data_loader.py

@@ -118,23 +118,8 @@ def transform_dataframe(df: pd.DataFrame, price_column: str):
 
     return result_df
 
-# def get_datasets(df: pd.DataFrame, training_minutes: int, pair: TradingPair) -> Tuple[pd.DataFrame, pd.DataFrame]:
-#     # Training dataset
-#     colname_a, colname_b = pair.colnames()
-#     df = df[["tstamp", colname_a, colname_b]]
-#     df = df.dropna()
 
-#     training_df = df.iloc[:training_minutes - 1, :].copy()
-#     training_df.reset_index(drop=True).dropna().reset_index(drop=True)
+# if __name__ == "__main__":
+#     df1 = load_sqlite_to_dataframe(sys.argv[1], table_name="md_1min_bars")
 
-#     # Testing dataset
-#     testing_df = df.iloc[training_minutes:, :].copy()
-#     testing_df.reset_index(drop=True).dropna().reset_index(drop=True)
-
-#     return (training_df, testing_df)
-
-
-if __name__ == "__main__":
-    df1 = load_sqlite_to_dataframe(sys.argv[1], table_name="md_1min_bars")
-
-    print(df1)
+#     print(df1)