main.py

from cpdbench.CPDBench import CPDBench
import cpdbench.examples.Functions as example_functions

cpdb = CPDBench()


@cpdb.dataset
def get_apple_dataset():
    return example_functions.dataset_get_apple_dataset()


@cpdb.dataset
def get_bitcoin_dataset():
    return example_functions.dataset_get_bitcoin_dataset()


# @cpdb.algorithm
# def execute_esst(signal):
#     return example_functions.algorithm_execute_single_esst(signal)


@cpdb.algorithm
def execute_esst_test(signal):
    return example_functions.algorithm_execute_single_esst(signal)


@cpdb.metric
def calc_accuracy(indexes, scores, ground_truth, *, window_size):
    return example_functions.metric_accuracy_in_allowed_windows(indexes, scores, ground_truth, window_size=window_size)


if __name__ == '__main__':
    cpdb.start("config.yml")