Feedback

×

• Feedback
• Was this page helpful?*
  • Yes
  • No
• Do you have any additional feedback?
  • Yes
  • No
• Additional Feedback

verticapy.machine_learning.memmodel.cluster.Clustering.predict

Clustering.predict(X: list | ndarray) → ndarray

Predicts clusters using the input matrix.

Parameters

X: ArrayLike

The data on which to make the prediction.

Returns

numpy.array

Predicted values.

Examples

Import the required module.

from verticapy.machine_learning.memmodel.cluster import KMeans

We will use the following attributes:

clusters = [[0.5, 0.6], [1, 2], [100, 200]]

p = 2

Let’s create a model.

model_km = KMeans(clusters, p)

Create a dataset.

data = [[2, 3]]

Compute the predictions.

model_km.predict(data)[0]
Out[6]: 1

Important

For this example, a specific model is utilized, and it may not correspond exactly to the model you are working with. To see a comprehensive example specific to your class of interest, please refer to that particular class.