def postprocess()

in optimum/amd/ryzenai/pipelines/image_classification.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

    def postprocess(self, model_outputs, top_k=5):
        outputs = model_outputs["logits"][0]
        outputs = outputs.numpy()

        scores = softmax(outputs)

        dict_scores = [{"label": i, "score": score.item()} for i, score in enumerate(scores)]

        dict_scores.sort(key=lambda x: x["score"], reverse=True)
        if top_k is not None:
            dict_scores = dict_scores[:top_k]

        return dict_scores