What problem does it solve?
Accelerates common machine learning workflows on tabular data by leveraging NVIDIA RAPIDS cuML to reduce training and inference time for large datasets, while providing a scikit-learn compatible API and sensible CPU fallbacks.
Core Features & Use Cases
- High-performance training: Train classification and regression models much faster on GPUs for datasets with tens of thousands of rows or more.
- Unsupervised learning & reduction: Run clustering (KMeans, DBSCAN), dimensionality reduction (PCA, UMAP) and visualization with GPU speedups.
- Preprocessing & compatibility: Includes guidance for feature scaling, label encoding, and float32 data requirements, and falls back to scikit-learn when GPUs or cuML are unavailable.
- Use case example: Rapidly prototype a RandomForestClassifier on a large customer dataset to detect churn, report accuracy, and produce ranked feature importances.
Quick Start
Train a GPU-accelerated RandomForestClassifier on data.csv predicting the target column and return dataset shape, train/test sizes, accuracy, and feature importances.