Update sklearn OHE

danielegrattarola · Jan 21, 2024 · bf60714 · bf60714
1 parent 386bf6f
commit bf60714
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spektral/datasets/tudataset.py b/spektral/datasets/tudataset.py
@@ -216,7 +216,7 @@ def _normalize(x, norm=None):
     Apply one-hot encoding or z-score to a list of node features
     """
     if norm == "ohe":
-        fnorm = OneHotEncoder(sparse=False, categories="auto")
+        fnorm = OneHotEncoder(sparse_output=False, categories="auto")
     elif norm == "zscore":
         fnorm = StandardScaler()
     else: