Kumo SDK

Kumo is an advanced machine learning (ML) platform that leverages Graph Neural Networks (GNNs) to deliver predictive analytics and insights directly from your relational data. Kumo simplifies the ML process by eliminating the need for extensive data preparation, allowing you to focus on generating valuable predictions and insights.

The Kumo Python SDK provides a composable, modular interface to train best-in-class GNNs on your relational data warehouse, powered by the Kumo machine learning platform.

If you haven’t already requested a demo or a free trial, please do so to receive an API key. If you already have an API key, get started with the Installation and Introduction by Example below!

Installation

• Installation
• API Key Provisioning and Management

Getting Started

• Introduction by Example

FAQ

• Working With Tables
• Working With Graphs
• Writing Predictive Queries
• Training Models and Generating Predictions

Package Reference

• kumoai
• kumoai.connector
• kumoai.graph
• kumoai.pquery
• kumoai.encoder
• kumoai.trainer
• kumoai.utils