Skip to content

GitLab

Closed
Created by Rosneft rosneft@rosneft_userDeveloper

Implement customizable prediction results for classification

In the current version the class probabilities in the only output of classification chains.

To ensure compatibility with external frameworks (like automlbenchmark):

  1. The class labels prediction should be implemented (e.g. predict instead predict_proba)
  2. The full probabilities option should be available for the binary classification (e.g. probs [0.2, 0.8] instead [0.8]).