Replace dtype to be more memory eficient

mirand863 · mirand863 · commit a9159f2e2efa · 2025-02-12T10:26:25.000-03:00
diff --git a/docs/source/api/utilities.rst b/docs/source/api/utilities.rst
@@ -95,13 +95,6 @@ F-score
 Datasets
 ----------
 
-Platypus diseases dataset
-^^^^^^^^^^^^^^^^^^^^^^^^^^
-
-.. autofunction:: datasets.load_platypus
-
-..................................
-
 Hierarchical text classification dataset
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
diff --git a/hiclass/datasets.py b/hiclass/datasets.py
@@ -82,7 +82,7 @@ def load_hierarchical_text_classification(test_size=0.3, random_state=42):
 
     data = [row for row in csv.reader(open(cached_file_path))]
     data.pop(0)
-    data = np.array(data)
+    data = np.array(data, dtype=object)
     X = data[:, 1]
     y = data[:, 7:]
 
