sktime / sktime

A unified framework for machine learning with time series

Home Page:https://www.sktime.net

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[BUG] classifiers failing on multiclass scenario due to `_get_train_probs`

fkiraly opened this issue · comments

The following classifiers fail on a scenario with three classes, see #6374 for the scenario:

  • Arsenal
  • BOSSEnsemble
  • ContractableBOSS
  • DrCIF
  • FreshPRINCE
  • ShapeletTransformClassifier
  • TemporalDictionaryEnsemble
  • TSCGridSearchCV - reason is not the classifier, but folds being too small in the scenario
  • WeightedEnsembleClassifier

A quick solution could be to introduce a capability:multiclass tag, but perhaps the bug is easy to fix.

Update: this is due to the _get_train_probs legacy method which has an undefined contract but which seems to assume a narrow range of data types (e.g., 3D numpy).