Merge pull request #39 from lorenzovolpi/ucimulti

Fix UCI multiclass features
HLT-ISTI · Jul 12, 2024 · 2140aed · 2140aed
2 parents 5da9fa0 + b543857
commit 2140aed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quapy/data/datasets.py b/quapy/data/datasets.py
@@ -751,7 +751,7 @@ def download(id, name):
         df = fetch_ucirepo(id=id)
 
         df.data.features = pd.get_dummies(df.data.features, drop_first=True)
-        X, y = df.data.features.to_numpy(), df.data.targets.to_numpy().squeeze()
+        X, y = df.data.features.to_numpy(dtype=np.float64), df.data.targets.to_numpy().squeeze()
 
         assert y.ndim == 1, 'more than one y'