---
warning: DO NOT CHANGE THIS MANUALLY, THIS IS GENERATED BY https://github.com/duckdb/community-extensions repository, check README there
title: ml
excerpt: |
  DuckDB Community Extensions
  Lightweight, columnar-native ML for DuckDB — train + inference, 34 algorithms, zero Python deps

extension:
  name: ml
  description: "Lightweight, columnar-native ML for DuckDB — train + inference, 34 algorithms, zero Python deps"
  version: 0.2.0
  language: Rust
  build: cargo
  license: MIT
  requires_toolchains: rust
  maintainers:
    - alitrack
  excluded_platforms: "wasm_eh;wasm_mvp;wasm_threads;windows_amd64_mingw;windows_amd64_rtools"

repo:
  github: alitrack/duckdb-ml
  ref: fd705cb580e15e4087e72db9587b3ec7d5748fe0
  ref_next: fd705cb580e15e4087e72db9587b3ec7d5748fe0

docs:
  hello_world: |
    -- Train linear regression
    SELECT * FROM ml_train('my_model', 'linear_regression',
      '[[1.0],[2.0],[3.0]]', '[[10.0],[20.0],[30.0]]', '{}');

    -- Predict
    SELECT ml_predict('my_model', 2.5);

  extended_description: |
    ## ML — Machine Learning for DuckDB

    Pure Rust ML extension. 34 algorithms, zero Python dependencies.

    **Algorithms (34):**
    - Regression: linear_regression, ridge_regression, lasso_regression, elastic_net, robust (Huber), polynomial_regression
    - Generalized Linear: logistic_regression, multilogistic, ordinal
    - Survival: cox, kaplan_meier
    - Trees: decision_tree, random_forest, rf_classifier, adaboost
    - Gradient Boosting: xgboost_regression, xgboost_binary (multi-class softmax), xgboost_regressor, xgboost_classifier
    - Neural: mlp_regressor
    - Distance: knn_regressor, knn_classifier
    - Bayesian: naive_bayes
    - Kernel: svm, svr
    - Clustering: kmeans, fuzzy_cmeans, dbscan, agglomerative
    - Dim Reduction: pca, lda, tsne
    - External: onnx, arima

    **Install:** `INSTALL ml FROM community;`

extension_star_count: 11
extension_star_count_pretty: 11
extension_download_count: 892
extension_download_count_pretty: 892
image: '/images/community_extensions/social_preview/preview_community_extension_ml.png'
layout: community_extension_doc
---
### Compatibility Matrix

This table covers all DuckDB releases that have not reached [end-of-life](https://duckdb.org/release_calendar), including LTS releases, and shows the 2 most recent patch releases of each. Older patch releases are not listed, but the extension may be available for them as well.

<div class="extension_compatibility_table"></div>

| Platform | 1.4.4 | 1.4.5 | 1.5.5 | 1.5.6 | 2.0.0 |
| :--- | :---: | :---: | :---: | :---: | :---: |
| Linux AMD64 | ❌ | ❌ | ✅ | ✅ | ✅ |
| Linux ARM64 | ❌ | ❌ | ✅ | ✅ | ✅ |
| macOS AMD64 | ❌ | ❌ | ✅ | ✅ | ✅ |
| macOS ARM64 | ❌ | ❌ | ✅ | ✅ | ✅ |
| Windows AMD64 | ❌ | ❌ | ✅ | ✅ | ✅ |
| Windows ARM64 | ❌ | ❌ | ❌ | ❌ | ❌ |
| WebAssembly | ❌ | ❌ | ❌ | ❌ | ❌ |

### Installing and Loading
```sql
INSTALL {{ page.extension.name }} FROM community;
LOAD {{ page.extension.name }};
```

{% if page.docs.hello_world %}
### Example
```sql
{{ page.docs.hello_world }}```
{% endif %}

{% if page.docs.extended_description %}
### About {{ page.extension.name }}
{{ page.docs.extended_description }}
{% endif %}
### Added Functions

<div class="extension_functions_table"></div>

|     function_name      | function_type | description | comment | examples |
|------------------------|---------------|-------------|---------|----------|
| ml_assoc_rules         | scalar        | NULL        | NULL    |          |
| ml_compare             | table         | NULL        | NULL    |          |
| ml_cox_train           | scalar        | NULL        | NULL    |          |
| ml_cross_validate      | scalar        | NULL        | NULL    |          |
| ml_deploy              | table         | NULL        | NULL    |          |
| ml_km_train            | scalar        | NULL        | NULL    |          |
| ml_list_models         | table         | NULL        | NULL    |          |
| ml_list_snapshots      | table         | NULL        | NULL    |          |
| ml_load_xgboost        | table         | NULL        | NULL    |          |
| ml_metrics             | scalar        | NULL        | NULL    |          |
| ml_ols                 | scalar        | NULL        | NULL    |          |
| ml_predict             | table         | NULL        | NULL    |          |
| ml_predict_batch       | table         | NULL        | NULL    |          |
| ml_predict_batch_value | scalar        | NULL        | NULL    |          |
| ml_similarity_value    | scalar        | NULL        | NULL    |          |
| ml_smote               | scalar        | NULL        | NULL    |          |
| ml_snapshot            | table         | NULL        | NULL    |          |
| ml_train               | table         | NULL        | NULL    |          |
| ml_train_model         | scalar        | NULL        | NULL    |          |
| ml_voting              | scalar        | NULL        | NULL    |          |

### Overloaded Functions

<div class="extension_functions_table"></div>

This extension does not add any function overloads.

### Added Types

<div class="extension_types_table"></div>

This extension does not add any types.

### Added Settings

<div class="extension_settings_table"></div>

|      name      |                       description                       | input_type | scope  | aliases | typed_value |
|----------------|---------------------------------------------------------|------------|--------|---------|-------------|
| temp_directory | Set the directory to which temporary files are written. | VARCHAR    | GLOBAL | []      | post.db.tmp |