scikit-learn/scikit-learn

Array API support for CalibratedClassifierCV

Open

#31.869 aberto em 2 de ago. de 2025

Ver no GitHub
 (15 comments) (1 reaction) (0 assignees)Python (27.020 forks)batch import
Array APIHardNew Featurehelp wantedmodule:calibration

Métricas do repositório

Stars
 (66.084 stars)
Métricas de merge de PR
 (Mesclagem média 10d) (90 fundiu PRs em 30d)

Description

Describe the workflow you want to enable

Towards #26024. Use CalibratedClassifierCV with pytorch or tensorflow models. This has become even more interesting use case with #31068.

Describe your proposed solution

In line with out Array API adoption path.

  • Temperature scaling
  • Platt scaling
  • Isotonic regression

Describe alternatives you've considered, if relevant

No response

Additional context

No response

Guia do colaborador

Direção de pesquisa
Implemente o suporte Array API para o escalonamento de Platt e a regressão isotônica no CalibratedClassifierCV, seguindo o padrão usado para o escalonamento de temperatura. Garanta a compatibilidade com os backends PyTorch e TensorFlow.
Pilha de tecnologia
python
Domain
machine learning
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
Pythonscikit learn
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
60