#31 · Primary category: Classical Machine Learning Frameworks
pecos
PECOS - Prediction for Enormous and Correlated Spaces
Project last updated:08/26/26
GitHub Stars
550
Forks
112
Contributors
32
License
Apache-2.0
Why we included this project
PECOS is built for a problem most ML libraries don't handle well: predicting from an output space that can reach hundreds of millions of candidates. It focuses on extreme multi-label ranking, where the task is to surface a small set of relevant labels from an enormous pool, and it offers two complementary approaches. X-Linear uses recursive linear models with fast C++ inference, while XR-Transformer fine-tunes pretrained transformers on multi-resolution objectives. The hierarchical label-tree design keeps both training and inference tractable at scale, which matters for teams doing product search, document tagging, or recommendation-style ranking. A bundled HNSW approximate nearest neighbor module handles sparse and dense features, so you get a practical retrieval tool alongside the ranking models.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
scikit-learn
scikit-learn: machine learning in Python
pyod
A Python library for anomaly detection across tabular, time series, graph, text, image, and audio data. 60+ detectors, benchmark-backed ADEngine orchestration, and an agentic workflow for AI agents.
umap
Uniform Manifold Approximation and Projection
dowhy
DoWhy is a Python library for causal inference that supports explicit modeling and testing of causal assumptions. DoWhy is based on a unified language for causal inference, combining causal graphical models and potential outcomes frameworks.
imbalanced-learn
A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning