A machine learning integrations library for TypeDB, enabling graph algorithms and Graph Neural Networks on strongly-typed graph data.
TypeDB-ML is a machine learning integrations library for TypeDB that enables graph algorithms and Graph Neural Networks (GNNs) on strongly-typed graph data. It provides tools to export TypeDB data to frameworks like NetworkX for graph analysis and PyTorch Geometric for building custom GNNs, solving the problem of applying advanced ML techniques to structured graph databases.
Data scientists and developers working with TypeDB who need to perform graph analysis, machine learning, or build predictive models on strongly-typed graph data, particularly those interested in Graph Neural Networks or graph algorithms.
Developers choose TypeDB-ML for its seamless integration with TypeDB's type system, support for heterogeneous data in PyTorch Geometric, and ready-to-use examples like link prediction, enabling ML on structured graph data without manual data conversion headaches.
TypeDB-ML is the Machine Learning integrations library for TypeDB
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Directly bridges TypeDB with ML frameworks, enabling seamless application of graph algorithms and GNNs on strongly-typed data without manual conversion, as highlighted in its philosophy.
Allows exporting TypeDB graph data to NetworkX with customizable query structures, providing access to a wide library of graph algorithms for analysis.
Includes a DataSet object for lazy loading graphs into PyG Data objects and utilities like store_concepts_by_type to handle heterogeneous data, preserving node ordering for GNNs.
Offers FeatureEncoder with encoders for continuous and categorical values, facilitating feature generation from TypeDB attributes for machine learning models.
The README explicitly states the repository is outdated and not supported, with plans to close by end of 2023, making it risky for production or new projects.
Tightly coupled with TypeDB, which has a smaller community compared to other graph databases, reducing peer support and extension availability.
Requires specific versions like TypeDB 2.11.1 and has intricacies in installing PyTorch Geometric dependencies, as noted in the quickstart section.