scm2cpp-learn: 44 scikit-learn estimators, same results, up to ~1000x faster

Описание и применение

scm2cpp-learn — набор из 44 заменителей scikit‑learn, которые дают те же результаты, но работают до 1000 раз быстрее и поддерживают GPU.

Задача

Python users face slow execution of scikit-learn estimators on large datasets and limited GPU support.

Результат

All 44 estimators produce coefficients matching scikit‑learn to rounding error while achieving up to 1,047× speed‑up on a single core and GPU acceleration on an RTX 4090.

Рабочий процесс

• pip install scm2cpp-learn
• replace import from sklearn with scm2cpp_learn
• instantiate estimator as usual
• fit and predict – same API and output as scikit‑learn

Доказательства

44 scikit-learn estimators (Lasso, LogisticRegression, RFE, PCA, SVM, ...) implementing the same algorithms with the same results — coefficients agree to rounding error

Источник

1.5x to ~1000x faster (e.g. RFE 1,047x; MultiTaskElasticNetCV ~100x on a single core)

Источник

scikit-learn API compatible: works in Pipeline and GridSearchCV, passes check_estimator

Источник