What problem does it solve?
This Skill reduces training and inference time for machine learning on large tabular datasets by leveraging NVIDIA cuML with a safe CPU fallback, making iterative modeling and experimentation practical on datasets where GPU acceleration yields significant speedups.
Core Features & Use Cases
- GPU-accelerated training and inference for common supervised and unsupervised algorithms (classification, regression, clustering).
- Compatibility and fallback: attempts cuDF/cuML and falls back to scikit-learn when GPU libraries are unavailable.
- Dimensionality reduction & preprocessing: PCA, UMAP (GPU only), scaling and label encoding for large feature sets.
- Use Case: Train a RandomForestClassifier on a 100k-row customer dataset with features cast to float32 to get rapid model iteration and feature importance ranking.
Quick Start
Use the cuml-machine-learning skill to train a GPU-accelerated model on your tabular dataset with numeric features cast to float32, fallback to scikit-learn if cuML is unavailable, and report train/test metrics.