DANet issue + dataset openml__arrhythmia__5

[duncanmcelfresh]

traceback:

Traceback (most recent call last):
  File "/home/shared/tabzilla/TabSurvey/tabzilla_experiment.py", line 136, in __call__
    result = cross_validation(model, self.dataset, self.time_limit)
  File "/home/shared/tabzilla/TabSurvey/tabzilla_utils.py", line 237, in cross_validation
    loss_history, val_loss_history = curr_model.fit(
  File "/home/shared/tabzilla/TabSurvey/models/danet.py", line 72, in fit
    self.model.fit(
  File "/home/shared/tabzilla/TabSurvey/models/danet_lib/abstract_model.py", line 125, in fit
    self.update_fit_params(X_train, y_train, eval_set)
  File "/home/shared/tabzilla/TabSurvey/models/danet_lib/DAN_Task.py", line 53, in update_fit_params
    check_output_dim(train_labels, y)
  File "/home/shared/tabzilla/TabSurvey/models/danet_lib/lib/multiclass_utils.py", line 349, in check_output_dim
    raise ValueError(
ValueError: Valid set -- {0, 1, 3, 4, 5, 6, 7, 8, 9, 11, 12} --
                             contains unkown targets from training --
                             {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12}