chronos-forecasting/scripts/experiments/synthetic_datasets/constants.py

NUM_YEARS = 5
START_DATE = "01/01/2024"
END_DATE = "12/31/2028"
NUM_EVENTS_PER_YEAR = 25
DATASETS_INFO = {
    "simple_spikes_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "function_target": "simple",
        "delta": 2,
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "simple_spikes_add": {
        "function_covariates": "steps",
        "op": "add",
        "delta": 2,
        "function_target": "simple",
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "diverse_spikes_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "delta": 2,
        "function_target": "diverse",
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "diverse_spikes_add": {
        "function_covariates": "steps",
        "op": "add",
        "delta": 2,
        "function_target": "diverse",
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "noisy_spikes_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "delta": 2,
        "function_target": "noisy",
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "noisy_spikes_add": {
        "function_covariates": "steps",
        "op": "add",
        "delta": 2,
        "function_target": "noisy",
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "simple_steps_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "function_target": "simple",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "simple_steps_add": {
        "function_covariates": "steps",
        "op": "add",
        "function_target": "simple",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "diverse_steps_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "function_target": "diverse",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "diverse_steps_add": {
        "function_covariates": "steps",
        "op": "add",
        "function_target": "diverse",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "noisy_steps_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "function_target": "noisy",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "noisy_steps_add": {
        "function_covariates": "steps",
        "op": "add",
        "function_target": "noisy",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "simple_bells_mult": {
        "function_covariates": "bells",
        "op": "mult",
        "function_target": "simple",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "simple_bells_add": {
        "function_covariates": "bells",
        "op": "add",
        "function_target": "simple",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "diverse_bells_mult": {
        "function_covariates": "bells",
        "op": "mult",
        "function_target": "diverse",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "diverse_bells_add": {
        "function_covariates": "bells",
        "op": "add",
        "function_target": "diverse",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "noisy_bells_mult": {
        "function_covariates": "bells",
        "op": "mult",
        "function_target": "noisy",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "noisy_bells_add": {
        "function_covariates": "bells",
        "op": "add",
        "function_target": "noisy",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "single_bells_mult": {
        "function_covariates": "bells",
        "op": "mult",
        "function_target": "single",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "single_bells_add": {
        "function_covariates": "bells",
        "op": "add",
        "function_target": "single",
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "single_steps_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "function_target": "single",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "single_steps_add": {
        "function_covariates": "steps",
        "op": "add",
        "function_target": "single",
        "delta": 15,
        "num_events": NUM_EVENTS_PER_YEAR * NUM_YEARS,
    },
    "single_spikes_mult": {
        "function_covariates": "steps",
        "op": "mult",
        "function_target": "single",
        "delta": 2,
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "single_spikes_add": {
        "function_covariates": "steps",
        "op": "add",
        "function_target": "single",
        "delta": 2,
        "num_events": 4 * NUM_EVENTS_PER_YEAR * NUM_YEARS,
        "fixed_event_pos": 15,
    },
    "single_arp_mult": {
        "function_covariates": "arp",
        "op": "mult",
        "function_target": "single",
    },
    "single_arp_add": {
        "function_covariates": "arp",
        "op": "add",
        "function_target": "single",
    },
    "simple_arp_mult": {
        "function_covariates": "arp",
        "op": "mult",
        "function_target": "simple",
    },
    "simple_arp_add": {
        "function_covariates": "arp",
        "op": "add",
        "function_target": "simple",
    },
    "diverse_arp_mult": {
        "function_covariates": "arp",
        "op": "mult",
        "function_target": "diverse",
    },
    "diverse_arp_add": {
        "function_covariates": "arp",
        "op": "add",
        "function_target": "diverse",
    },
    "noisy_arp_mult": {
        "function_covariates": "arp",
        "op": "mult",
        "function_target": "noisy",
    },
    "noisy_arp_add": {
        "function_covariates": "arp",
        "op": "add",
        "function_target": "noisy",
    },
}