Merge pull request #78 from DynamicsAndNeuralSystems/jmoo2880-norm-detrend-fix

norm detrend fix

Author: joshuabmoore

.github/SECURITY.md

@@ -15,4 +15,5 @@ currently being supported with security updates.
 | ------- | ------------------ |
 | 1.1.0   | :white_check_mark: |
 | 1.1.1   | :white_check_mark: |
+| 2.0.0   | :white_check_mark: |
 

.github/workflows/run_dataset_generation.yaml

@@ -29,7 +29,7 @@ jobs:
         run: |
           python tests/generate_benchmark_tables.py
       - name: Upload artifact
-        uses: actions/upload-artifact@v3
+        uses: actions/upload-artifact@v4
         with:
           name: benchmark-tables
           path: tests/CML7_benchmark_tables_new.pkl

README.md

@@ -70,7 +70,7 @@ Once you have installed _pyspi_, you can learn how to apply the package by check
 - [Finance: stock price time series](https://time-series-features.gitbook.io/pyspi/usage/walkthrough-tutorials/finance-stock-price-time-series)
 
 
-- [Neuroimaging: fMRI time series)](https://time-series-features.gitbook.io/pyspi/usage/walkthrough-tutorials/neuroimaging-fmri-time-series)
+- [Neuroimaging: fMRI time series](https://time-series-features.gitbook.io/pyspi/usage/walkthrough-tutorials/neuroimaging-fmri-time-series)
 
 ### Advanced Usage
 For advanced users, we offer several additional guides in the [full documentation](https://time-series-features.gitbook.io/pyspi/usage/advanced-usage) on how you can distribute your _pyspi_ jobs across PBS clusters, as well as how you can construct your own subsets of SPIs. 

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyspi"
-version = "1.1.1"
+version = "2.0.0"
 authors = [
     { name ="Oliver M. Cliff", email="[email protected]"},
 ]

pyspi/calculator.py

@@ -5,6 +5,8 @@
 from tqdm import tqdm
 from collections import Counter
 from scipy import stats
+from colorama import init, Fore
+init(autoreset=True)
 
 # From this package
 from .data import Data
@@ -34,18 +36,22 @@ class Calculator:
             A pre-configured subset of SPIs to use. Options are "all", "fast", "sonnet", or "fabfour", defaults to "all".
         configfile (str, optional):
             The location of the YAML configuration file for a user-defined subset. See :ref:`Using a reduced SPI set`, defaults to :code:`'</path/to/pyspi>/pyspi/config.yaml'`
+        detrend (bool, optional):
+            If True, detrend the dataset along the time axis before normalising (if enabled), defaults to True.
         normalise (bool, optional):
-            Normalise the dataset along the time axis before computing SPIs, defaults to True.
+            If True, z-score normalise the dataset along the time axis before computing SPIs, defaults to True.
+            Detrending (if enabled) is always applied before normalisation.
     """
     _optional_dependencies = None
 
     def __init__(
         self, dataset=None, name=None, labels=None, subset="all", configfile=None,
-        normalise=True
+        detrend=True, normalise=True
     ):
         self._spis = {}
         self._excluded_spis = list()
         self._normalise = normalise
+        self._detrend = detrend
 
         # Define configfile by subset if it was not specified
         if configfile is None:
@@ -89,11 +95,11 @@ def __init__(
         self._labels = labels
 
         print(f"="*100)
-        print(f"Number of SPIs: {len(self.spis)}\n")
+        print(Fore.GREEN + f"{len(self.spis)} SPI(s) were successfully initialised.\n")
         if len(self._excluded_spis) > 0:
             missing_deps = [dep for dep, is_met in self._optional_dependencies.items() if not is_met]
-            print("**** SPI Initialisation Warning ****")
-            print("\nSome dependencies were not detected, which has led to the exclusion of certain SPIs:")
+            print(Fore.YELLOW + "**** SPI Initialisation Warning ****")
+            print(Fore.YELLOW + "\nSome dependencies were not detected, which has led to the exclusion of certain SPIs:")
             print("\nMissing Dependencies:")
 
             for dep in missing_deps:
@@ -115,7 +121,7 @@ def __init__(
                     print(f"  - {spi}")
 
             print(f"\n" + "="*100)
-            print("\nOPTIONS TO PROCEED:\n")
+            print(Fore.YELLOW + "\nOPTIONS TO PROCEED:\n")
             print(f"  1) Install the following dependencies to access all SPIs: [{', '.join(missing_deps)}]")
             callable_name = "{Calculator/CalculatorFrame}"
             print(f"  2) Continue with a reduced set of {self.n_spis} SPIs by calling {callable_name}.compute(). \n")
@@ -256,7 +262,7 @@ def load_dataset(self, dataset):
                 New dataset to attach to calculator.
         """
         if not isinstance(dataset, Data):
-            self._dataset = Data(Data.convert_to_numpy(dataset), normalise=self._normalise)
+            self._dataset = Data(Data.convert_to_numpy(dataset), normalise=self._normalise, detrend=self._detrend)
         else:
             self._dataset = dataset
 
@@ -297,7 +303,7 @@ def compute(self):
                 warnings.warn(f'Caught {type(err)} for SPI "{spi}": {err}')
                 self._table[spi] = np.nan
         pbar.close()
-        print(f"\nCalculation complete. Time taken: {pbar.format_dict['elapsed']:.4f}s")
+        print(Fore.GREEN + f"\nCalculation complete. Time taken: {pbar.format_dict['elapsed']:.4f}s")
         inspect_calc_results(self)
 
     def _rmmin(self):
@@ -505,7 +511,7 @@ def init_from_list(self, datasets, names, labels, **kwargs):
             self.add_calculator(calc)
 
     def init_from_yaml(
-        self, document, normalise=True, n_processes=None, n_observations=None, **kwargs
+        self, document, detrend=True, normalise=True, n_processes=None, n_observations=None, **kwargs
     ):
         datasets = []
         names = []
@@ -524,6 +530,7 @@ def init_from_yaml(
                             data=file,
                             dim_order=dim_order,
                             name=names[-1],
+                            detrend=detrend,
                             normalise=normalise,
                             n_processes=n_processes,
                             n_observations=n_observations,

pyspi/data.py

@@ -7,10 +7,11 @@
 from pyspi import utils
 from scipy.stats import zscore
 from scipy.signal import detrend
+from colorama import init, Fore
 import os
 
 VERBOSE = False
-
+init(autoreset=True) # automatically reset coloured outputs
 
 class Data:
     """Store data for dependency analysis.
@@ -40,8 +41,11 @@ class Data:
             2-dimensional array with raw data, defaults to None.
         dim_order (str, optional):
             Order of dimensions, accepts two combinations of the characters 'p', and 's' for processes and observations, defaults to 'ps'.
+        detrend (bool, optional):
+            If True, detrend the dataset along the time axis before normalising (if enabled), defaults to True.
         normalise (bool, optional):
-            If True, data is z-scored (normalised) along the time dimension, defaults to True.
+            If True, z-score normalise the dataset along the time axis before computing SPIs, defaults to True.
+            Detrending (if enabled) is always applied before normalisation.
         name (str, optional):
             Name of the dataset
         procnames (list, optional):
@@ -57,13 +61,15 @@ def __init__(
         self,
         data=None,
         dim_order="ps",
+        detrend=True,
         normalise=True,
         name=None,
         procnames=None,
         n_processes=None,
         n_observations=None,
     ):
         self.normalise = normalise
+        self.detrend = detrend
         if data is not None:
             dat = self.convert_to_numpy(data)
             self.set_data(
@@ -176,15 +182,20 @@ def set_data(
         if n_observations is not None:
             data = data[:, :n_observations]
 
-        if self.normalise:
-            print("Normalising the dataset...\n")
-            data = zscore(data, axis=1, nan_policy="omit", ddof=1)
+        if self.detrend:
+            print(Fore.GREEN + "[1/2] De-trending the dataset...")
             try:
                 data = detrend(data, axis=1)
             except ValueError as err:
                 print(f"Could not detrend data: {err}")
         else:
-            print("Skipping normalisation of the dataset...\n")
+            print(Fore.RED + "[1/2] Skipping detrending of the dataset...")
+
+        if self.normalise:
+            print(Fore.GREEN + "[2/2] Normalising (z-scoring) the dataset...\n")
+            data = zscore(data, axis=1, nan_policy="omit", ddof=1)
+        else:
+            print(Fore.RED + "[2/2] Skipping normalisation of the dataset...\n")
 
         nans = np.isnan(data)
         if nans.any():

pyspi/utils.py

@@ -4,6 +4,8 @@
 import pandas as pd
 import os
 import yaml 
+from colorama import Fore, init
+init(autoreset=True)
 
 def _contains_nan(a, nan_policy='propagate'):
     policies = ['propagate', 'raise', 'omit']
@@ -230,6 +232,10 @@ def filter_spis(keywords, output_name = None, configfile= None):
 """)
 
 def inspect_calc_results(calc):
+    """
+    Display a summary of the computed SPI results, including counts of successful computations, 
+    outputs with NaNs, and partially computed results.
+    """
     total_num_spis = calc.n_spis
     num_procs = calc.dataset.n_processes
     spi_results = dict({'Successful': list(), 'NaNs': list(), 'Partial NaNs': list()})

requirements.txt

@@ -21,3 +21,4 @@ tslearn
 mne==0.23.0
 seaborn
 future
+colorama

setup.py

@@ -29,7 +29,8 @@
         'tslearn',
         'mne==0.23.0',
         'seaborn',
-        'future'
+        'future',
+        'colorama'
 ]
 
 testing_extras = [
@@ -63,7 +64,7 @@
                         'data/standard_normal.npy',
                         'data/cml7.npy']},
     include_package_data=True,
-    version='1.1.1',
+    version='2.0.0',
     description='Library for pairwise analysis of time series data.',
     author='Oliver M. Cliff',
     author_email='[email protected]',