classification_dataset bool

tsfm_public/toolkit/util.py

@@ -626,6 +626,7 @@ class values.
                     has_class_labels_tag = True
                     class_label_list = [token.strip() for token in tokens[2:]]
                     metadata_started = True
+                    classification_dataset = True
                 elif line.startswith("@targetlabel"):
                     if data_started:
                         raise IOError("metadata must come before data")
@@ -647,7 +648,6 @@ class values.
                         )
                     has_class_labels_tag = True
                     metadata_started = True
-                    regression = True
                 # Check if this line contains the start of data
                 elif line.startswith("@data"):
                     if line != "@data":
@@ -705,7 +705,7 @@ class values.
                                     # Check if we have reached a class label
                                     if line[char_num] != "(" and class_labels:
                                         class_val = line[char_num:].strip()
-                                        if not regression:
+                                        if classification_dataset:
                                             if class_val not in class_label_list:
                                                 raise IOError(
                                                     "the class value '"