Map promoted oneHotEncoding

Author: FlorentinD

graphdatascience/tests/integration/test_util_ops.py

@@ -103,6 +103,13 @@ def test_util_nodeProperty(gds: GraphDataScience, G: Graph) -> None:
     assert result == 1337
 
 
-def test_ml_oneHotEncoding(gds: GraphDataScience) -> None:
+@pytest.mark.filterwarnings("ignore: .*gds.alpha.ml.oneHotEncoding.*")
+def test_alpha_ml_oneHotEncoding(gds: GraphDataScience) -> None:
     result = gds.alpha.ml.oneHotEncoding(["Chinese", "Indian", "Italian"], ["Italian"])
     assert result == [0, 0, 1]
+
+
+@pytest.mark.compatible_with(min_inclusive=ServerVersion(2, 24, 0))
+def test_util_oneHotEncoding(gds: GraphDataScience) -> None:
+    result = gds.util.oneHotEncoding(["Chinese", "Indian", "Italian"], ["Italian"])
+    assert result == [0, 0, 1]

graphdatascience/utils/util_proc_runner.py

@@ -1,5 +1,7 @@
 from typing import Any
 
+from graphdatascience.call_parameters import CallParameters
+
 from ..error.illegal_attr_checker import IllegalAttrChecker
 from ..error.uncallable_namespace import UncallableNamespace
 from ..utils.util_node_property_func_runner import NodePropertyFuncRunner
@@ -43,3 +45,25 @@ def asNodes(self, node_ids: list[int]) -> list[Any]:
     @property
     def nodeProperty(self) -> NodePropertyFuncRunner:
         return NodePropertyFuncRunner(self._query_runner, self._namespace + ".nodeProperty", self._server_version)
+
+    def oneHotEncoding(self, available_values: list[Any], selected_values: list[Any]) -> list[int]:
+        """
+        One hot encode a list of values.
+
+        Args:
+            available_values: The available values to encode.
+            selected_values: The values to encode.
+
+        Returns:
+            The one hot encoded values.
+        """
+        namespace = self._namespace + ".oneHotEncoding"
+
+        params = CallParameters(
+            available_values=available_values,
+            selected_values=selected_values,
+        )
+        query = f"RETURN {namespace}($available_values, $selected_values) AS encoded"
+        result = self._query_runner.run_cypher(query=query, params=params)
+
+        return result.iat[0, 0]  # type: ignore