.. _ai-api:

AI
====

graphistry['ai'] provides a set of utilities for AI and machine learning workflows on graphs, with optional GPU support

.. toctree::
   :maxdepth: 2


Featurize 
---------
.. automodule:: graphistry.feature_utils
    :members:
    :undoc-members:
    :inherited-members:
    :show-inheritance:

.. autodata:: graphistry.feature_utils.FeatureEngine

.. autodata:: graphistry.feature_utils.FeatureEngineConcrete

UMAP 
----
.. automodule:: graphistry.umap_utils
    :members:
    :undoc-members:
    :inherited-members:
    :show-inheritance:

Semantic Search 
---------------
.. automodule:: graphistry.text_utils
    :members:
    :undoc-members:
    :inherited-members:
    :show-inheritance:

DBSCAN
------
.. automodule:: graphistry.compute.cluster
    :members:
    :undoc-members:
    :inherited-members:
    :show-inheritance:

RGCN
----
.. automodule:: graphistry.networks
    :members:
    :undoc-members:
    :show-inheritance:

HeterographEmbedModuleMixin
---------------------------
.. automodule:: graphistry.embed_utils
    :members:
    :undoc-members:
    :show-inheritance: