def METHOD_NAME(self, predict_scores):
    if self.drop_label:
        # dropping NONE to get binary predictions from 3-way classifier
        # TODO maybe this should be model dependent rather than task dependent
        label_list=np.array(self.label_list)
        mask=label_list==self.drop_label
        masked_predict_scores=predict_scores.copy()
        masked_predict_scores[:,mask]=-9e19
    else:
            masked_predict_scores=predict_scores
    predictions = np.argmax(masked_predict_scores, axis=1)
    return predictions