def test_make_catalogue():
if os.path.exists('test_catalogueconstructor'):
shutil.rmtree('test_catalogueconstructor')
dataio = DataIO(dirname='test_catalogueconstructor')
localdir, filenames, params = download_dataset(name='olfactory_bulb')
dataio.set_data_source(type='RawData', filenames=filenames, **params)
#~ dataio.set_manual_channel_group(range(14))
dataio.set_manual_channel_group([5, 6, 7, 8, 9])
catalogueconstructor = CatalogueConstructor(dataio=dataio)
catalogueconstructor.set_preprocessor_params(chunksize=1024,
#signal preprocessor
highpass_freq=300,
backward_chunksize=1280,
#peak detector
peak_sign='-', relative_threshold=7, peak_span=0.0005,
)
t1 = time.perf_counter()
catalogueconstructor.estimate_signals_noise(seg_num=0, duration=10.)
t2 = time.perf_counter()
print('estimate_signals_noise', t2-t1)
t1 = time.perf_counter()
catalogueconstructor.run_signalprocessor()
t2 = time.perf_counter()
print('run_signalprocessor', t2-t1)
print(catalogueconstructor)
t1 = time.perf_counter()
catalogueconstructor.extract_some_waveforms(n_left=-12, n_right=15, nb_max=10000)
t2 = time.perf_counter()
print('extract_some_waveforms', t2-t1)
# PCA
t1 = time.perf_counter()
catalogueconstructor.project(method='pca', n_components=12, batch_size=16384)
t2 = time.perf_counter()
print('project', t2-t1)
# cluster
t1 = time.perf_counter()
catalogueconstructor.find_clusters(method='kmeans', n_clusters=13)
t2 = time.perf_counter()
print('find_clusters', t2-t1)
# trash_small_cluster
catalogueconstructor.trash_small_cluster()
catalogueconstructor.make_catalogue()
def debug_interp_centers0():
dataio = DataIO(dirname='test_catalogueconstructor')
catalogueconstructor = CatalogueConstructor(dataio=dataio)
catalogue = catalogueconstructor.make_catalogue()
centers = catalogue['centers0']
interp_centers = catalogue['interp_centers0']
subsample_ratio = catalogue['subsample_ratio']
