import numpy as np
from aeon.transformations.series.hidalgo import Hidalgo
np.random.seed(123)
X = np.random.rand(100, 3)
X[:60, 1:] += 10
X[60:, 1:] = 0
model = Hidalgo(K=2, burn_in=0.8, n_iter=1000, seed=10)
fitted_model = model.fit(X)
Z = fitted_model.transform(X)
Z
array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], dtype=int64)