Skip to content

Commit e459782

Browse files
committed
feat: Add time-series model configuration to YAML configs
Introduces a new `ts_models` section in both `config_hyperopt.yml` and `config_single_run.yml`. This allows for granular control over which time-series classification algorithms (such as Rocket, MUSE, ResNet, etc.) are instantiated when the pipeline is running in `time_series_mode`. Changes include: - `config_hyperopt.yml`: Added a comprehensive list of aeon-based time-series models with default enabled/disabled states. Documented known issues for disabled models (e.g., HIVECOTEV2, InceptionTime). - `config_single_run.yml`: Added a subset of time-series models (Rocket, TimeSeriesForest, KNeighbors) for single-run testing and debugging.
1 parent 9509682 commit e459782

2 files changed

Lines changed: 37 additions & 0 deletions

File tree

config_hyperopt.yml

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,36 @@ models:
7171
AutoKerasClassifierClass: false # disabled by default
7272

7373

74+
# Time-series models to be used during the hyperopt search.
75+
# These are only loaded if time_series_mode is enabled in the pipeline.
76+
ts_models:
77+
# Enabled by default
78+
RocketClassifier: true
79+
TimeSeriesForestClassifier: true
80+
KNeighborsTimeSeriesClassifier: true
81+
Catch22Classifier: true
82+
HIVECOTEV2: false # Disabled due to internal numba typing error in aeon library
83+
TSFreshClassifier: true
84+
Arsenal: true
85+
CNNClassifier: true
86+
ElasticEnsemble: true
87+
EncoderClassifier: true
88+
FCNClassifier: true
89+
FreshPRINCEClassifier: true
90+
InceptionTimeClassifier: false # Disabled due to Keras input shape mismatch error
91+
IndividualInceptionClassifier: false # Disabled due to Keras input shape mismatch error
92+
IndividualTDE: true
93+
MLPClassifier: true
94+
MUSE: true
95+
OrdinalTDE: true
96+
ResNetClassifier: true
97+
SignatureClassifier: true
98+
SummaryClassifier: true
99+
TemporalDictionaryEnsemble: true
100+
# Univariate models, disabled by default
101+
ContractableBOSS: false
102+
HIVECOTEV1: false
103+
74104
# This section defines the parameter search space for Hyperopt.
75105
# The structure uses lists of options, which will be parsed into hp.choice.
76106
hyperopt_search_space:

config_single_run.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,13 @@ models:
5252
FLAMLClassifierClass: false
5353
AutoKerasClassifierClass: false
5454

55+
# Time-series models for a single run.
56+
# These are only loaded if time_series_mode is enabled in the pipeline.
57+
ts_models:
58+
RocketClassifier: true
59+
TimeSeriesForestClassifier: false
60+
KNeighborsTimeSeriesClassifier: false
61+
5562
# This section defines a single set of parameters for a standalone run.
5663
run_params:
5764
outcome_var_n: 1

0 commit comments

Comments
 (0)