astrocp Library Brings Class-Conditional Conformal Prediction to Astronomical Surveys
A developer named Pedro Sordo Martínez has released astrocp (AD-MCP), an open-source Python library designed to provide class-conditional prediction coverage for astronomical datasets containing rare object classes, such as supernovae. The library addresses a gap in existing tools by applying Mondrian stratification via anomaly scores from IsolationForest, rather than relying on a single global conformity quantile, which can mask poor coverage for underrepresented classes. Built on top of MAPIE 1.4.1 and scikit-learn, it uses a RandomForest base model and cross-validated lambda selection to control prediction set construction. Testing on SDSS and PLAsTiCC survey datasets showed improved conditional coverage relative to a global conformal baseline in favorable regimes, though performance depends on having sufficient per-class calibration samples. The repository is publicly available on GitHub under the AGPL-3.0-or-later license, with reproducibility verified through a clean-clone audit that also uncovered and fixed three environment-related bugs.
This is an AI-generated summary. ShortSingh links to the original source for the complete article.
Discussion (0)
Log in to join the discussion and vote.
Log in