Skip to content

Instantly share code, notes, and snippets.

@quantra-go-algo

quantra-go-algo/cuml_example_01.py

Created March 6, 2024 17:54
Show Gist options
  • Save quantra-go-algo/2d3d8732cd8194907bec022a6fac9f63 to your computer and use it in GitHub Desktop.
Save quantra-go-algo/2d3d8732cd8194907bec022a6fac9f63 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cuml_example_01.py
# Import the libraries
from cuml.cluster import KMeans
import cudf
# Generate sample data
data = cudf.DataFrame()
data['feature1'] = cp.random.rand(1000)
data['feature2'] = cp.random.rand(1000)
# Perform KMeans clustering
kmeans = KMeans(n_clusters=3)
kmeans.fit(data)
labels = kmeans.predict(data)
print(labels)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment