def predict_many()

in submission_code/infer.py [0:0]

• 20 lines of code
• 6 McCabe index (conditional complexity)

    def predict_many(self, texts, result_cnt):
        
        alpha = 0.6
        n_utils = 5
        beam_width = 5
        
        text_cleaned = [clean_text(x) for x in texts]
        
        pred_utils = self.util_model.predict_many(text_cleaned, n_utils)
        
        result = []
        with torch.no_grad():
            for i in range(len(text_cleaned)):
                candidates = []
                for j in range(n_utils):
                    util, util_proba = pred_utils[i][j]
                    pred = self.ctx_model.predict(text_cleaned[i], util, beam_width)

                    for pred_cmd, ctx_proba in pred:
                        joined_proba = (1 - alpha) * util_proba + alpha * ctx_proba
                        candidates.append((pred_cmd,joined_proba))
                        
                
                candidates = sorted(candidates, key=lambda x: -x[1])[:result_cnt]
                candidates = [x[0] for x in candidates]
                result.append(candidates)
                
        return result