dbt-infer

The Infer adapter plugin for dbt


License
Apache-2.0
Install
pip install dbt-infer==1.6.0

Documentation

Infer logo dbt logo

Unit Tests Badge

dbt-infer

The dbt adapter for Infer. dbt-infer allows you to connect your dbt instance to Infer and use SQL-inf syntax in your dbt models.

dbt-infer acts as a layer between your existing data warehouse and your dbt instance, enabling you to perform ML analytics within your dbt models.

dbt-infer packages are hosted on PyPi.

SQL-inf

SQL-inf is an extension of SQL that introduces machine learning primitives to SQL. These primitives can be used within any part of your SQL queries, or DBT models, and allow you to build ML analytics use cases.

Read more about SQL-inf here.

Examples

Illustrative example based on an idealised table users with some simple demographic data, whether the user has churned or not, their lifetime value(LTV) and a text field with customer feedback.

Predict column has_churn from the other columns in the table users.

SELECT * FROM users PREDICT(has_churned)

Understand what columns drive values of has_churned

SELECT * FROM users EXPLAIN(PREDICT(has_churned))

Predict and understand the LTV in column ltv from table users.

SELECT * FROM users PREDICT(ltv)
SELECT * FROM users EXPLAIN(PREDICT(ltv))

Perform text analysis, sentiment and topic analysis, on user feedback

SELECT * FROM users SENTIMENT(feedback)
SELECT * FROM users TOPICS(feedback)

Create user segmentations on demographics

SELECT age, location, gender, job, education FROM users CLUSTER()

Find the sizes of the user segmentations

SELECT cluster_id, COUNT(*) as size FROM (
    SELECT age, location, gender, job, education FROM users CLUSTER()
) GROUP BY cluster_id

Find users similar to the user with user_id=123

SELECT * FROM users SIMILAR_TO(user_id=123)

Analyse what, if any, demographic features drive feedback sentiment

SELECT age, location, gender, job, education, prediction FROM (
    SELECT * FROM users SENTIMENT(feedback)
) EXPLAIN(PREDICT(prediction))

Installation

Detailed installation and setup guide here.

Requirements

You should be using dbt 1.2 or later.

Setup Infer account

First you need to setup your Infer account and generate your api key.

Read about how to do that here.

Install dbt-infer

pip install dbt-infer

Setting up dbt-infer

Setup a target in your profile for dbt-infer with the following shape

<target_name>:
  url: <infer-api-endpoint>
  username: <infer-api-username>
  apikey: <infer-apikey>
  type: infer
  data_config:
    <here goes your normal data warehouse config>

where data_config contains the profile settings for your underlying data warehouse.