code: def probs_to_labels(prediction: np.array):
      """Converts predicted probabilities into labels"""
      list_with_labels = []
      for list_with_probs in prediction:
          list_with_labels.append(list_with_probs.argmax())
  
      return np.asarray(list_with_labels).reshape((-1, 1))
  
imports: import math
  import os
  import platform
  import random
  import tempfile
  from pathlib import Path
  from typing import Optional, Union
  import numpy
  import pandas
  from golem.utilities.random import RandomStateHandler
  from sklearn.model_selection import train_test_split
label_desc: Use image node_id 4 for calling a global function with example usage: probs_to_labels(prediction) and returns: np