From c85f35df5a834d73fd598ce57abbe9d03e6ca062 Mon Sep 17 00:00:00 2001
From: Peter Luenenschloss <peter.luenenschloss@ufz.de>
Date: Mon, 12 Oct 2020 12:40:50 +0200
Subject: [PATCH] added option to set changepoints flags n cpd cluster func
---
saqc/funcs/modelling.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/saqc/funcs/modelling.py b/saqc/funcs/modelling.py
index 3c48457f8..38d2dbc95 100644
--- a/saqc/funcs/modelling.py
+++ b/saqc/funcs/modelling.py
@@ -432,7 +432,7 @@ def _reduceCPCluster(stat_arr, thresh_arr, start, end, obj_func, num_val):
@register(masking='field')
def modelling_clusterByChangePoints(data, field, flagger, stat_func, thresh_func, bwd_window, min_periods_bwd,
fwd_window=None, min_periods_fwd=None, closed='both', try_to_jit=True,
- reduce_window=None, reduce_func=lambda x, y: x.argmax(), **kwargs):
+ reduce_window=None, reduce_func=lambda x, y: x.argmax(), flag_changepoints=False, **kwargs):
"""
Assigns label to the data, aiming to reflect continous regimes of the processes the data is assumed to be
generated by.
@@ -536,4 +536,6 @@ def modelling_clusterByChangePoints(data, field, flagger, stat_func, thresh_func
cluster = cluster.cumsum()
data[field] = cluster
flagger = flagger.setFlags(field, flag=flagger.UNFLAGGED, force=True, **kwargs)
+ if flag_changepoints:
+ flagger.setFlags(field, loc=det_index)
return data, flagger
\ No newline at end of file
--
GitLab