def predict_classification()

in Project-AutoML/automl/mod/mod_autosklearn.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

    def predict_classification(self, inputs):

        inputs = self.oridinal_encoder.transform(inputs)

        pred_proba = self.classifier.predict_proba(inputs)
        label_indexes = pred_proba.argmax(axis=1)
        probs = pred_proba[np.arange(pred_proba.shape[0]), label_indexes]
        labels = (
            self.classifier.automl_.InputValidator.target_validator.inverse_transform(
                label_indexes
            )
        )
        result = []
        for label, pro in zip(labels, probs):
            result.append({"label": label, "confidence": float(pro)})
        return result