def predict_classification()

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

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

    def predict_classification(self, inputs):
        pred_proba = self.automl.predict_proba(inputs)
        label_indexes = pred_proba.argmax(axis=1)
        probs = pred_proba[np.arange(pred_proba.shape[0]), label_indexes]
        labels = self.automl._label_transformer.inverse_transform(
            pd.Series(label_indexes.astype(int))
        )
        result = []
        for label, pro in zip(labels, probs):
            result.append({"label": label, "confidence": float(pro)})
        return result