Hello, I am creating a KNN classifier from scratch. Below is a code segment, is there any way to rewrite it differently? I am using Python. Thank you.

Select K neighbors Prediction Metrics def prediction_classification(train_x, train_y, test_row, num_neighbors): output_neighbors, output_labels, output_distances = get_neighbors(train_x, train_y, test_row, num_neighbors) label_cnt = np.bincount(output_labels) prediction = np.argmax(Glabel_cnt)] return prediction argmax def argmax (a, axis=None, out=None)