All functions

cuML_major_version()

Get the major version of the RAPIDS cuML shared library cuda.ml was linked to.

cuML_minor_version()

Get the minor version of the RAPIDS cuML shared library cuda.ml was linked to.

cuda.ml

cuda.ml

cuda_ml_agglomerative_clustering()

Perform Single-Linkage Agglomerative Clustering.

cuda_ml_can_predict_class_probabilities()

Determine whether a CuML model can predict class probabilities.

cuda_ml_dbscan()

Run the DBSCAN clustering algorithm.

cuda_ml_fil_enabled()

Determine whether Forest Inference Library (FIL) functionalities are enabled in the current installation of cuda.ml.

cuda_ml_fil_load_model()

Load a XGBoost or LightGBM model file.

cuda_ml_inverse_transform()

Apply the inverse transformation defined by a trained cuML model.

cuda_ml_is_classifier()

Determine whether a CuML model is a classifier.

cuda_ml_kmeans()

Run the K means clustering algorithm.

cuda_ml_knn()

Build a KNN model.

cuda_ml_knn_algo_ivfflat()

Build a specification for the "ivfflat" KNN query algorithm.

cuda_ml_knn_algo_ivfpq()

Build a specification for the "ivfpq" KNN query algorithm.

cuda_ml_knn_algo_ivfsq()

Build a specification for the "ivfsq" KNN query algorithm.

cuda_ml_lasso()

Train a linear model using LASSO regression.

cuda_ml_ols()

Train a OLS model.

cuda_ml_pca()

Perform principal component analysis.

cuda_ml_rand_forest()

Train a random forest model.

cuda_ml_rand_proj()

Random projection for dimensionality reduction.

cuda_ml_ridge()

Train a linear model using ridge regression.

cuda_ml_serialize() cuda_ml_serialise()

Serialize a CuML model

cuda_ml_svm()

Train a SVM model.

cuda_ml_transform()

Transform data using a trained cuML model.

cuda_ml_tsne()

t-distributed Stochastic Neighbor Embedding.

cuda_ml_tsvd()

Truncated SVD.

cuda_ml_umap()

Uniform Manifold Approximation and Projection (UMAP) for dimension reduction.

cuda_ml_unserialize() cuda_ml_unserialise()

Unserialize a CuML model state

has_cuML()

Determine whether cuda.ml was linked to a valid version of the RAPIDS cuML shared library.

predict(<cuda_ml_fil>)

Make predictions on new data points.

predict(<cuda_ml_knn>)

Make predictions on new data points.

predict(<cuda_ml_linear_model>)

Make predictions on new data points.

predict(<cuda_ml_rand_forest>)

Make predictions on new data points.

predict(<cuda_ml_svm>)

Make predictions on new data points.