v0.0.4

AlexanderVNikitin · Dec 15, 2023 · 11ded2d · 11ded2d
1 parent 7f284b2
commit 11ded2d
Show file tree

Hide file tree

Showing 4 changed files with 57 additions and 87 deletions.
diff --git a/README.md b/README.md
@@ -37,7 +37,7 @@ pip install tsgm
 
 - [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1l2VB6eUwvrxyu8iB30faGiQM5AKthc82?usp=sharing) Introductory Tutorial "[Getting started with TSGM](https://github.com/AlexanderVNikitin/tsgm/blob/main/tutorials/GANs/cGAN.ipynb)"
 - [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1Vw9t4TlI1Nek_t6bMPyKcPPPqCiXfOK3?usp=sharing) Tutorial on using [Time Series Augmentations](https://github.com/AlexanderVNikitin/tsgm/blob/main/tutorials/augmentations.ipynb)
-- Tutorial on [Evaluation of Synthetic Time Series Data](https://github.com/AlexanderVNikitin/tsgm/blob/main/tutorials/evaluation.ipynb)
+- [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1hubtddSX94KyLzuCTwmU6pAFBgBeiEB-?usp=sharing) Tutorial on [Evaluation of Synthetic Time Series Data](https://github.com/AlexanderVNikitin/tsgm/blob/main/tutorials/evaluation.ipynb)
 - [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1wpf9WeNVj5TkUcPF6EavVx-hUCOfyvUd?usp=sharing) Tutorial on using [Multiple GPUs or TPU with TSGM](https://github.com/AlexanderVNikitin/tsgm/blob/main/tutorials/Using%20Multiple%20GPUs%20or%20TPU.ipynb)
 
 For more examples, see [our tutorials](./tutorials).

diff --git a/setup.py b/setup.py
@@ -44,7 +44,7 @@ def read_file(filename: str) -> str:
 
 
 setup(name='tsgm',
-      version='0.0.3',
+      version='0.0.4',
       description='Time Series Generative Modelling Framework',
       author=author,
       author_email='',
@@ -66,6 +66,7 @@ def read_file(filename: str) -> str:
           "seaborn",
           "scikit-learn",
           "prettytable",
+          "antropy==0.1.6",
           "yfinance==0.2.28",
           "tqdm",
           "dtaidistance >= 2.3.10",

diff --git a/tsgm/metrics/metrics.py b/tsgm/metrics/metrics.py
@@ -275,7 +275,7 @@ def __call__(self, d_hist: tsgm.dataset.DatasetOrTensor, d_syn: tsgm.dataset.Dat
         X_all, y_all = np.concatenate([X_hist, X_syn]), np.concatenate([[1] * len(d_hist), [0] * len(d_syn)])
         X_train, X_test, y_train, y_test = sklearn.model_selection.train_test_split(X_all, y_all, test_size=test_size, random_state=random_seed)
         model.fit(X_train, y_train, epochs=n_epochs)
-        y_pred = model.predict(X_test)
+        y_pred = (model.predict(X_test) > 0.5).astype(int)
         if metric is None:
             return sklearn.metrics.accuracy_score(y_test, y_pred)
         else:

diff --git a/tutorials/evaluation.ipynb b/tutorials/evaluation.ipynb