---
title: Clustering Analysis and Dimensionality Reduction
slug: d7ZJnIyNuZJEHaoovs3VM
description: Discover clustering and dimensionality reduction methods in Ocient with OcientML for grouping data and simplifying analysis.
docTags: 
createdAt: Fri Mar 01 2024 23:30:52 GMT+0000 (Coordinated Universal Time)
updatedAt: Mon Mar 30 2026 18:52:29 GMT+0000 (Coordinated Universal Time)
---

This tutorial uses examples to explain the clustering and dimensionality reduction capabilities in OcientML™.&#x20;

## Clustering Models

Clustering models bear similarities to classification, but they use unsupervised learning, which means they do not use any class labels. Instead, the algorithm tries to identify groupings on its own by finding clusters of data that seem closer to each other and farther away from other clusters.&#x20;

These models generate an integer label, but its ordering is arbitrary. The Ocient® clustering models require you to specify upfront the number of clusters.&#x20;

### K-Means Clustering

K-means is by far the most well-known clustering algorithm because it is simple and fast. K-means performs particularly well if you can scale your features so clusters are roughly circular and equal in size.&#x20;

These examples demonstrate the k-means model. The examples do not use a data set that is primed for optimal performance of the model, such that they show the shortcomings of the k-means model and how it can perform relatively well even with sub-optimal data. &#x20;

The examples use a data set of three-dimensional points.

```sql
SELECT x, y, z FROM mldemo.clusters_3d LIMIT 10;
x                     y                     z
------------------------------------------------------------------
-2.332818550827427    2.1009728159912044    -0.7199553405333495
0.27417054580238553   -0.29545777905624226  0.6797340230215586
-2.8608875606526794   2.8532647291218884    -0.7296054608643955
-4.9539478834915345   -4.987358010702081    -0.818608017693332
0.7043958454569171    -0.7209720558364705   -0.7879961167668906
4.949298606341526     4.956911476243746     0.6015700102502752
2.245935331812275     -2.158347482865157    -1.0786578947921444
-1.3725012757407247   1.5652243270484643    -2.5951429274728794
4.516051945094492     4.560705560578962     -0.1569348000058415
5.140105808358035     -4.9583101530224      -1.0064835641873153

Fetched 10 rows
```

This example shows a k-means model created over this data. You can specify multiple options, but the only one that is required is the `k` value, which represents the number of clusters.

```sql
CREATE MLMODEL kmeans1 TYPE KMEANS ON (
    SELECT x,
        y,
        z
    FROM mldemo.clusters_3d
) options('k'->'3');

Modified 0 rows
```

This data set has labels, but they are hidden from the model to allow for unsupervised learning.&#x20;

These example queries compare how well the unsupervised k-means clustering performs against the actual correct classes and labels.&#x20;

Before you can do that, you must figure out which cluster numbers correspond to which labels.

```sql
SELECT COUNT(*),
    cluster_num
FROM (
        SELECT kmeans1(x, y, z) AS cluster_num
        FROM mldemo.clusters_3d
        WHERE label = 'LEFT'
    )
GROUP BY cluster_num;

count(*)            cluster_num
--------------------------------
3165                0
146329              1

Fetched 2 rows

SELECT COUNT(*),
    cluster_num
FROM (
        SELECT kmeans1(x, y, z) AS cluster_num
        FROM mldemo.clusters_3d
        WHERE label = 'RIGHT'
    )
GROUP BY cluster_num;

count(*)            cluster_num
--------------------------------
3877                0
145761              2

Fetched 2 rows

SELECT COUNT(*),
    cluster_num
FROM (
        SELECT kmeans1(x, y, z) AS cluster_num
        FROM mldemo.clusters_3d
        WHERE label = 'CENTER'
    )
GROUP BY cluster_num;

count(*)            cluster_num
--------------------------------
700868              0

Fetched 1 row
```

The results indicate that cluster `0` is the `CENTER` class, cluster `1` is `LEFT`, and cluster `2` is `RIGHT`.&#x20;

With that information, you can compute the overall accuracy.

```sql
SELECT count(*) / 1000000.0
FROM mldemo.clusters_3d
WHERE (
        kmeans1(x, y, z) = 0
        AND label = 'CENTER'
    )
    OR (
        kmeans1(x, y, z) = 1
        AND label = 'LEFT'
    )
    OR (
        kmeans1(x, y, z) = 2
        AND label = 'RIGHT'
    );
    
(_count(*)_0)/((1000000.0))
----------------------------
0.992958

Fetched 1 row
```

The accuracy is very good. This calculation indicates that the model never miscategorized `CENTER`, but there were some wrong classifications for the `LEFT` and `RIGHT` labels, although they were rare.&#x20;

For details, see [K-Means Clustering](<./././SQL Reference/Machine Learning Model Functions/Clustering and Dimension Reduction Models.md>).

### Gaussian Mixture Models

Gaussian mixture models (GMMs) also perform clustering, however they use a significantly more complex algorithm. GMMs can handle several things that k-means cannot, such as:&#x20;

- GMMs can handle clusters that are not circular, i.e., they have different variances in different directions.
- GMMs can handle clusters that have an arbitrary rotation, i.e., they can have covariances.
- GMMs can handle the fact that all clusters might not be equally as likely, i.e. if a point is located right between two clusters, then it is more likely to be the one more common in the training data.
- GMMs can show the probability of a new point belonging to each cluster rather than outputting a single cluster value.

GMMs operate by finding a weighted mixture of k multi-variate Gaussians that is most likely to represent the population from which the data was sampled. Each Gaussian in the mix has a mean vector, which represents the center of each cluster.&#x20;

After you create the model, it is simple to determine the cluster that a point most likely belongs to. The examples demonstrate how to find the highest probability cluster.&#x20;

This example makes a model over the same data set. This model requires the `numDistributions` option, which represents the number of clusters.

```sql
CREATE MLMODEL gmm TYPE GAUSSIAN MIXTURE MODEL ON (
    SELECT x,
        y,
        z
    FROM mldemo.clusters_3d
) options('metrics'->'true', 'numDistributions'->'3');

Modified 0 rows
```

This query executes the GMM model.

```sql
SELECT gmm(0,0,0) AS class_probabilities;

class_probabilities
--------------------------------------------------------------------------------
[[2.0445296213545875E-4, 0.999582295498657, 2.1325153920763012E-4]]

Fetched 1 row
```

The results are very different than executing a k-means model because these are probabilities. In this case, the output means the probability of `(0,0,0)` being the second class is very high (greater than 99.9 percent), while the probability of it being the other classes is essentially zero.&#x20;

To find the most likely class, use the `VECTOR_ARGMAX` function, which returns a 1-based class index.&#x20;

```sql
SELECT VECTOR_ARGMAX(gmm(0,0,0)) AS class_probabilities;

class_probabilities
--------------------
1

Fetched 1 row
```

You can write a query to assess the accuracy of this model. But first, you must figure out the association between classes and labels.

```sql
SELECT COUNT(*),
    cluster_num
FROM (
        SELECT VECTOR_ARGMAX(gmm(x, y, z)) AS cluster_num
        FROM mldemo.clusters_3d
        WHERE label = 'LEFT'
    )
GROUP BY cluster_num;

count(*)            cluster_num
----------------------------------------
149483              3
2                   1
9                   2

Fetched 3 rows

SELECT COUNT(*),
    cluster_num
FROM (
        SELECT VECTOR_ARGMAX(gmm(x, y, z)) AS cluster_num
        FROM mldemo.clusters_3d
        WHERE label = 'CENTER'
    )
GROUP BY cluster_num;

count(*)            cluster_num
----------------------------------------
700868              1

Fetched 1 row

SELECT COUNT(*),
    cluster_num
FROM (
        SELECT VECTOR_ARGMAX(gmm(x, y, z)) AS cluster_num
        FROM mldemo.clusters_3d
        WHERE label = 'RIGHT'
    )
GROUP BY cluster_num;

count(*)            cluster_num
----------------------------------------
6                   2
3                   3
149629              1

Fetched 3 rows
```

This model misclassified 20 rows, which is far better than the 7,042 rows the k-means model misclassified. This is a direct result of the additional complexity of the GMM.

GMM models can handle much more complex situations than k-means models, but this comes at the cost of more time training and executing the model. The time a GMM model takes compared to a k-means model depends on the number of clusters.

For details, see [Gaussian Mixture](<./././SQL Reference/Machine Learning Model Functions/Clustering and Dimension Reduction Models.md>).

## Dimensionality Reduction

Dimensionality reduction algorithms reduce the number of input features while still keeping as much of the meaningful properties of the data as possible. Models are quicker to build, and often higher quality after a dimensionality reduction model reduces the number of input features.&#x20;

A common first step in an analysis is to use dimensionality reduction to simplify the data. See these examples that demonstrate how to use dimensionality reduction as a preprocessing step before using other model types.&#x20;

### Principal Component Analysis&#x20;

Principal component analysis (PCA) is an unsupervised algorithm that only operates on the inputs, and does not understand what the data is being used for. It is also a linear dimensionality reduction algorithm, meaning that the new features it generates are linear combinations of existing features.

PCA generates as many new features as there are input features. So by itself, it is not reducing the number of dimensions. However, PCA creates new features that try to maximize variance and sorts the new features in terms of the amount of variance they contain. The system catalog tables provide information on how much variance is contained by the new features. You can use this information to find how many trailing new features to drop.

This PCA tutorial uses a new data set. Start with a regression problem that is trying to find the best-fit polynomial for `f(x1, x2, x3) = y`. The tutorial starts with a [Polynomial Regression](<./././SQL Reference/Machine Learning Model Functions/Regression Models.md>) model to see how well it does using three input features, and then it uses PCA to reduce the number of variables without losing accuracy.

This example shows the Polynomial Regression model.&#x20;

```sql
CREATE MLMODEL poly TYPE POLYNOMIAL REGRESSION ON (
    SELECT x1,
        x2,
        x3,
        y
    FROM mldemo.pca_poly
) options('order'->'2', 'metrics'->'true');

Modified 0 rows
```

The system catalog table shows that the model fits the data perfectly.&#x20;

```sql
SELECT coefficient_of_determination
FROM sys.machine_learning_models a,
    sys.polynomial_regression_models b
WHERE a.id = b.machine_learning_model_id
    AND name = 'poly';
    
coefficient_of_determination
-----------------------------
1.0

Fetched 1 row
```

In this instance, PCA can reduce the three independent variables to two.

The first step is to build a PCA model over the input features.

```sql
CREATE MLMODEL pca1 TYPE PRINCIPAL COMPONENT ANALYSIS ON (
    SELECT x1,
        x2,
        x3
    FROM mldemo.pca_poly
);

Modified 0 rows
```

The model does not include `y` because PCA operates over only the input features.&#x20;

Examine the `machine_learning_models` and `principal_component_analysis_models` system catalog tables.

```sql
SELECT importance
FROM sys.machine_learning_models a,
    sys.principal_component_analysis_models b
WHERE a.id = b.machine_learning_model_id
    AND name = 'pca1';
    
importance
--------------------------------------------------------------------------------
[0.6022962127628679, 0.33286202821081917, 0.06484175902631285]

Fetched 1 row
```

The `importance` value indicates that over 93 percent of the signal is in the first two PCA output features, which means it is possible to have a robust model even if you remove the third feature.

This example uses the Polynomial Regression model to access the PCA output features. The example executes the `pca1` PCA function with the PCA input features followed by the number of features. This number starts at `1`, so this example uses PCA features `1` and `2`, but not the last feature.

```sql
CREATE MLMODEL poly2 TYPE POLYNOMIAL REGRESSION ON (
    SELECT pca1(x1, x2, x3, 1),
        pca1(x1, x2, x3, 2),
        y
    FROM mldemo.pca_poly
) options('order'->'2', 'metrics'->'true');

Modified 0 rows
```

Notice that the example still references all three input features, but this model now has only two independent variables. As a result, the model trains 15 percent faster than the version with three independent variables. It is also a much simpler model. The model has six terms in the polynomial regression instead of 10, when there were three independent variables.&#x20;

This query examines how well the model fits the data.

```sql
SELECT coefficient_of_determination
FROM sys.machine_learning_models a,
    sys.polynomial_regression_models b
WHERE a.id = b.machine_learning_model_id
    AND name = 'poly2';
    
coefficient_of_determination
-----------------------------
0.9952839146818417

Fetched 1 row
```

The model is a great fit with more than 99 percent accuracy. &#x20;

In contrast, here is an example where PCA is not a good idea. This example goes back to the three-dimensional clusters data set used in the [Clustering Models](./#clustering-models) examples.&#x20;

This example creates the PCA model and checks the `importance` value of the PCA output features in the `machine_learning_models` and `principal_component_analysis_models` system catalog tables.

```sql
CREATE MLMODEL pca2 TYPE PRINCIPAL COMPONENT ANALYSIS ON (
    SELECT x,
        y,
        z
    FROM mldemo.clusters_3d
);

Modified 0 rows

SELECT importance
FROM sys.machine_learning_models a,
    sys.principal_component_analysis_models b
WHERE a.id = b.machine_learning_model_id
    AND name = 'pca2';
    
importance
--------------------------------------------------------------------------------
[0.3688781607553818, 0.3333390832624229, 0.2977827559821952]

Fetched 1 row
```

In this case, all three PCA output features are fairly evenly weighted. By removing the last feature, the model covers only about 70% of the signal. This strongly indicates that proceeding with a two-feature model would be problematic and inaccurate.&#x20;

### Linear Discriminant Analysis

While PCA is unsupervised, linear discriminant analysis (LDA) is a supervised dimensionality reduction algorithm. LDA works only with numeric classification algorithms, but the classification can be binary or multi-class. Hence, LDA understands how the data should be used, making it operate differently.

Creating an LDA model is mostly similar to PCA. One difference is that LDA also requires a class or label.

This tutorial uses the three-dimensional cluster data that the PCA model struggled to work with. The example uses LDA to reduce the features of the cluster data by using the label already contained in the data set.&#x20;

```sql
CREATE MLMODEL lda1 TYPE LINEAR DISCRIMINANT ANALYSIS ON (
    SELECT x,
        y,
        z,
        label
    FROM mldemo.clusters_3d
);

Modified 0 rows
```

As with PCA, there is an `importance` column in the system catalog tables.

```sql
SELECT importance
FROM sys.machine_learning_models a,
    sys.linear_discriminant_analysis_models b
WHERE a.id = b.machine_learning_model_id
    AND name = 'lda1';
    
importance
--------------------------------------------------------------------------------
[1.0000000000000133, -1.4051695910079908E-20, -1.3342566035320652E-14]

Fetched 1 row
```

This output strongly indicates that only the first LDA output feature matters, which means that it is possible to make a single feature classification model.&#x20;

This example takes the LDA-reduced model as input and uses it to create a neural network model, using the `FEEDFORWARD NETWORK` model. Note that this model uses the `cross_entropy_loss` option to enable multi-class classification. &#x20;

```sql
CREATE MLMODEL three_d_clusters_with_1_feature TYPE FEEDFORWARD NETWORK ON (
    SELECT lda1(x, y, z, 1),
        CASE
            WHEN label = 'LEFT' THEN { { 1.0,
            0.0,
            0.0 } }
            WHEN label = 'RIGHT' THEN { { 0.0,
            1.0,
            0.0 } }
            ELSE { { 0.0,
            0.0,
            1.0 } }
        END AS target
    FROM mldemo.clusters_3d
) options(
    'metrics'->'true',
    'hiddenLayers'->'2',
    'hiddenLayerSize'->'4',
    'outputs'->'3',
    'lossFunction'->'cross_entropy_loss',
    'useSoftMax'->'true'
);

Modified 0 rows
```

&#x20;Execute the model for a new point to see if it returns a reasonable value.&#x20;

```sql
SELECT three_d_clusters_with_1_feature(lda1(0, 0, 0, 1)) AS predicted;

predicted
--------------------------------------------------------------------------------
[[0.007386760165583984, 0.0022085315302674473, 0.9904047083041486]]

Fetched 1 row
```

In this case, the query asks for a prediction for the input `0, 0, 0`. This input first passes through the LDA model because that is what the neural network model was built on.&#x20;

The query asks the LDA model for its first component, and then runs the neural network on that value. The neural network outputs a vector of class probabilities; `VECTOR_ARGMAX` returns the 1‑based index of the highest probability.&#x20;

In this example, the highest probability class is the third class, which the query defines as `CENTER`.

```sql
SELECT count(*) / 1000000.0
FROM mldemo.clusters_3d
WHERE (
        VECTOR_ARGMAX(
            three_d_clusters_with_1_feature(lda1(x, y, z, 1))
        ) = 1
        AND label = 'LEFT'
    )
    OR (
        VECTOR_ARGMAX(
            three_d_clusters_with_1_feature(lda 1(x, y, z, 1))
        ) = 2
        AND label = 'RIGHT'
    )
    OR (
        VECTOR_ARGMAX(
            three_d_clusters_with_1_feature(lda1(x, y, z, 1))
        ) = 3
        AND label = 'CENTER'
    );
    
(_count(*)_0)/((1000000.0))
----------------------------
0.818108

Fetched 1 row
```

This model is nearly 82 percent accurate despite being simplified from three features to one. While the accuracy is worse, the time required to train this model is greatly reduced.&#x20;

## Related Links

[Machine Learning Model Functions](<./././SQL Reference/Machine Learning Model Functions.md>)&#x20;

[Clustering and Dimension Reduction Models](<./././SQL Reference/Machine Learning Model Functions/Clustering and Dimension Reduction Models.md>)

