def test_scrimp_plus_plus():
ts = np.array([0, 0, 1, 0, 0, 0, 1, 0])
m = 4
step_size = 0.25
profile = scrimp.scrimp_plus_plus(ts,
m,
step_size=step_size,
sample_pct=1.0)
expected_mp = np.array([0, 3.2660, 3.2660, 3.2660, 0])
expected_mpidx = np.array([
4,
2,
0,
0,
0,
])
np.testing.assert_almost_equal(profile['mp'], expected_mp, decimal=4)
np.testing.assert_equal(profile['pi'], expected_mpidx)
ts = np.loadtxt(os.path.join(MODULE_PATH, '..', 'tests', 'sampledata.txt'))
m = 32
step_size = 0.25
profile = scrimp.scrimp_plus_plus(ts,
m,
step_size=step_size,
sample_pct=1.0)
expected_mp = np.loadtxt(
os.path.join(MODULE_PATH, '..', 'tests', 'scrimp.mp.txt'))
expected_mpi = np.loadtxt(
os.path.join(MODULE_PATH, '..', 'tests',
'scrimp.mpi.txt')).astype('int') - 1
np.testing.assert_almost_equal(profile['mp'], expected_mp)
np.testing.assert_equal(profile['pi'], expected_mpi)
def compute(ts, windows=None, query=None, sample_pct=1, threshold=0.98,
n_jobs=1):
"""
Computes the exact or approximate MatrixProfile based on the sample percent
specified. Currently, MPX and SCRIMP++ is used for the exact and
approximate algorithms respectively. When multiple windows are passed, the
Pan-MatrixProfile is computed and returned.
By default, only passing in a time series (ts), the Pan-MatrixProfile is
computed based on the maximum upper window algorithm with a correlation
threshold of 0.98.
Note
----
When multiple windows are passed and the Pan-MatrixProfile is computed, the
query is ignored!
Parameters
----------
ts : array_like
The time series to analyze.
windows : int or array_like
The window(s) to compute the MatrixProfile. Note that it may be an int
for a single matrix profile computation or an array of ints for
computing the pan matrix profile.
query : array_like, Optional
The query to analyze. Note that when computing the PMP the query is
ignored!
sample_pct : float, default = 1
A float between 0 and 1 representing how many samples to compute for
the MP or PMP. When it is 1, the exact algorithm is used.
threshold : float, Default 0.98
The correlation coefficient used as the threshold. It should be between
0 and 1. This is used to compute the upper window size when no
window(s) is given.
n_jobs : int, Default = 1
Number of cpu cores to use.
Returns
-------
dict : profile
The profile computed.
"""
result = None
multiple_windows = core.is_array_like(windows) and len(windows) > 1
no_windows = isinstance(windows, type(None))
has_threshold = isinstance(threshold, float)
if no_windows and not has_threshold:
raise ValueError('compute requires a threshold or window(s) to be set!')
if core.is_array_like(windows) and len(windows) == 1:
windows = windows[0]
# compute the upper window and pmp
if no_windows and has_threshold:
profile = maximum_subsequence(ts, threshold, include_pmp=True)
# determine windows to be computed
# from 8 in steps of 2 until upper w
start = 8
windows = range(start, profile['upper_window'] + 1)
# compute the pmp
result = skimp(ts, windows=windows, sample_pct=sample_pct,
pmp_obj=profile)
# compute the pmp
elif multiple_windows:
if core.is_array_like(query):
logger.warn('Computing PMP - query is ignored!')
result = skimp(ts, windows=windows, sample_pct=1,
n_jobs=n_jobs)
# compute exact mp
elif sample_pct >= 1:
result = mpx(ts, windows, query=query, n_jobs=n_jobs)
# compute approximate mp
else:
result = scrimp_plus_plus(ts, windows, query=query, n_jobs=n_jobs,
sample_pct=sample_pct)
return result
def test_invalid_random_state_exception():
exc = 'Invalid random_state value given.'
with pytest.raises(ValueError) as excinfo:
scrimp.scrimp_plus_plus([1, 2, 3, 4, 5], 2, random_state='adsf')
assert exc in str(excinfo.value)
def test_invalid_step_size_greater():
exc = 'step_size should be a float between 0 and 1.'
with pytest.raises(ValueError) as excinfo:
scrimp.scrimp_plus_plus([1, 2, 3, 4, 5], 2, 2)
assert exc in str(excinfo.value)
def test_window_size_minimum_exception():
with pytest.raises(ValueError) as excinfo:
scrimp.scrimp_plus_plus([1, 2, 3, 4, 5], 2, 0.25)
assert 'Window size must be at least 4' in str(excinfo.value)
def test_time_series_too_short_exception():
with pytest.raises(ValueError) as excinfo:
scrimp.scrimp_plus_plus([1, 2, 3, 4, 5], 4, 0.25)
assert 'Time series is too short' in str(excinfo.value)
def compute(ts,
windows=None,
query=None,
sample_pct=1,
threshold=0.98,
n_jobs=1,
preprocessing_kwargs=None):
"""
Computes the exact or approximate MatrixProfile based on the sample percent
specified. Currently, MPX and SCRIMP++ is used for the exact and
approximate algorithms respectively. When multiple windows are passed, the
Pan-MatrixProfile is computed and returned.
By default, only passing in a time series (ts), the Pan-MatrixProfile is
computed based on the maximum upper window algorithm with a correlation
threshold of 0.98.
Notes
-----
When multiple windows are passed and the Pan-MatrixProfile is computed, the
query is ignored!
Parameters
----------
ts : array_like
The time series to analyze.
windows : int, array_like
The window(s) to compute the MatrixProfile. Note that it may be an int
for a single matrix profile computation or an array of ints for
computing the pan matrix profile.
query : array_like, optional
The query to analyze. Note that when computing the PMP the query is
ignored!
sample_pct : float, default 1
A float between 0 and 1 representing how many samples to compute for
the MP or PMP. When it is 1, the exact algorithm is used.
threshold : float, default 0.98
The correlation coefficient used as the threshold. It should be between
0 and 1. This is used to compute the upper window size when no
window(s) is given.
n_jobs : int, default = 1
Number of cpu cores to use.
preprocessing_kwargs : dict, default = None
A dictionary object to sets parameters for preprocess function.
A valid preprocessing_kwargs should have the following structure:
>>> {
>>> 'window': The window size to compute the mean/median/minimum/maximum value,
>>> 'method': A string indicating the data imputation method, which should be
>>> 'mean', 'median', 'min' or 'max',
>>> 'direction': A string indicating the data imputation direction, which should be
>>> 'forward', 'fwd', 'f', 'backward', 'bwd', 'b'. If the direction is
>>> forward, we use previous data for imputation; if the direction is
>>> backward, we use subsequent data for imputation.,
>>> 'add_noise': A boolean value indicating whether noise needs to be added into the
>>> time series
>>> }
To disable preprocessing procedure, set the preprocessing_kwargs to
None/False/""/{}.
Returns
-------
dict : profile
The profile computed.
"""
result = None
multiple_windows = core.is_array_like(windows) and len(windows) > 1
no_windows = isinstance(windows, type(None))
has_threshold = isinstance(threshold, float)
if no_windows and not has_threshold:
raise ValueError(
'compute requires a threshold or window(s) to be set!')
# Check to make sure all window sizes are greater than 3, return a ValueError if not.
if (isinstance(windows, int)
and windows < 4) or (multiple_windows
and np.any(np.unique(windows) < 4)):
raise ValueError(
'Compute requires all window sizes to be greater than 3!')
if core.is_array_like(windows) and len(windows) == 1:
windows = windows[0]
# preprocess the time series
preprocessing_kwargs = validate_preprocess_kwargs(preprocessing_kwargs)
if preprocessing_kwargs:
ts = preprocess(
ts,
window=preprocessing_kwargs['window'],
impute_method=preprocessing_kwargs['impute_method'],
impute_direction=preprocessing_kwargs['impute_direction'],
add_noise=preprocessing_kwargs['add_noise'])
# compute the upper window and pmp
if no_windows and has_threshold:
profile = maximum_subsequence(ts, threshold, include_pmp=True)
# determine windows to be computed
# from 8 in steps of 2 until upper w
start = 4
#start = 8
windows = range(start, profile['upper_window'] + 1)
# compute the pmp
result = skimp(ts,
windows=windows,
sample_pct=sample_pct,
pmp_obj=profile)
# compute the pmp
elif multiple_windows:
if core.is_array_like(query):
logger.warn('Computing PMP - query is ignored!')
result = skimp(ts, windows=windows, sample_pct=1, n_jobs=n_jobs)
# compute exact mp
elif sample_pct >= 1:
result = mpx(ts, windows, query=query, n_jobs=n_jobs)
# compute approximate mp
else:
result = scrimp_plus_plus(ts,
windows,
query=query,
n_jobs=n_jobs,
sample_pct=sample_pct)
return result