Added a clause for when the target attribute class is capitalized. Should probably reload it so reloading the ARFF file a second time isnt necessary, but this is a quick fix to make the code work for a course currently given at TUe.

Author: PGijsbers

openml/datasets/dataset.py

@@ -249,6 +249,8 @@ def _retrieve_class_labels(self):
         dataAttributes = dict(arffData['attributes'])
         if('class' in dataAttributes):
             return dataAttributes['class']
+        elif('Class' in dataAttributes):
+            return dataAttributes['Class']
         else:
             return None