ucl-pond / pySuStaIn

Subtype and Stage Inference (SuStaIn) algorithm with an example using simulated data.
MIT License
135 stars 64 forks source link

Added Zscore Missing Data version #18

Closed marestarellas closed 3 years ago

marestarellas commented 3 years ago

Created new file to deal with missing data in Zscore SuStaIn. Check _calculate_likelihood_stage for change in code.

sea-shunned commented 3 years ago

Hi Mar,

Can you please modify the current __init__.py file in the main pySuStaIn folder to also add the missing score version, i.e. adding from .ZScoreSustainMissingData import *.

sea-shunned commented 3 years ago

Local tests (without missing data) passed fine, so happy to merge once the above is done.

Later, we will need to adjust the test for the ZScore class for this version to give it missing data as input, but that's a separate/future issue.