feat(ts): Evaluation added.

ydataai · Jan 26, 2021 · e7ca201 · e7ca201
1 parent 4a24c01
commit e7ca201
Show file tree

Hide file tree

Showing 2 changed files with 322 additions and 12 deletions.
diff --git a/examples/timeseries/TimeGAN_Synthetic_stock_data.ipynb b/examples/timeseries/TimeGAN_Synthetic_stock_data.ipynb
diff --git a/src/ydata_synthetic/preprocessing/timeseries/stock.py b/src/ydata_synthetic/preprocessing/timeseries/stock.py
@@ -23,7 +23,10 @@ def transformations(seq_len: int):
         # Reading the stock data
         stock_df = pd.read_csv('../data/stock.csv')
 
-    stock_df = stock_df.set_index('Date').sort_index()
+    try:
+        stock_df = stock_df.set_index('Date').sort_index()
+    except:
+        stock_df=stock_df
     #Data transformations to be applied prior to be used with the synthesizer model
     processed_data = real_data_loading(stock_df.values, seq_len=seq_len)