def test_Kmeans_init():
"""Test Kmeans"""
covset = generate_cov(20, 3)
labels = np.array([0, 1]).repeat(10)
# init
km = Kmeans(2)
# fit
km.fit(covset)
# fit with init
km = Kmeans(2, init=covset[0:2])
km.fit(covset)
# fit with labels
km.fit(covset, y=labels)
# predict
km.predict(covset)
# transform
km.transform(covset)
# n_jobs
km = Kmeans(2, n_jobs=2)
km.fit(covset)
def test_Kmeans_transform():
"""Test transform of Kmeans"""
covset = generate_cov(20, 3)
km = Kmeans(2)
km.fit(covset)
km.transform(covset)
def test_Kmeans_fit_parallel():
"""Test Fit of Kmeans using paralell"""
covset = generate_cov(20, 3)
km = Kmeans(2, n_jobs=2)
km.fit(covset)
def test_Kmeans_predict():
"""Test prediction of Kmeans"""
covset = generate_cov(20, 3)
km = Kmeans(2)
km.fit(covset)
km.predict(covset)
def test_Kmeans_fit_with_y():
"""Test Fit of Kmeans with a given y"""
covset = generate_cov(20, 3)
labels = np.array([0, 1]).repeat(10)
km = Kmeans(2)
km.fit(covset, y=labels)
def test_Kmeans_fit_with_init():
"""Test Fit of Kmeans wit matric initialization"""
covset = generate_cov(20, 3)
km = Kmeans(2, init=covset[0:2])
km.fit(covset)
def test_Kmeans_fit():
"""Test Fit of Kmeans"""
covset = generate_cov(20, 3)
km = Kmeans(2)
km.fit(covset)
def test_Kmeans_init():
"""Test init of Kmeans"""
km = Kmeans(2)
labels_base = copy.deepcopy(labels)
# Covariance Matrix transorm
off_cov_matrix = Covariances(estimator='lwf').transform(off_epochs_data)
# MDM model init and fit
mdm = MDM(metric=dict(mean='riemann', distance='riemann'))
mdm.fit(off_cov_matrix, labels)
# End of offline training
# EEG stream on the lab network
print("looking for an EEG stream...")
streams = resolve_stream('name', 'openvibeSignal')
# Create a new inlet to read from the stream
inlet = StreamInlet(streams[0])
kmeans = Kmeans(n_clusters=4)
time_window = timeWindowInit(inlet)
time_window_base = copy.deepcopy(time_window)
timeBase = time.time()
count = 0
time_array = []
while not keyboard.is_pressed('s'):
time_window = copy.deepcopy(time_window_base)
while time_window.shape[1] < 769:
sample, timestamp = inlet.pull_sample()
sample = np.array(sample)
time_window = np.column_stack((time_window, sample))
actualTime = time.time() - timeBase
