code: def round(self, value_array):
          
  
          rounded_values = np.zeros(value_array.shape)
          rounded_values[np.argmax(value_array)] = 1
          return rounded_values
docs: Rounds a categorical variable by setting to one the max of the given vector and to zero the rest of the entries.
          Assumes an 1x[number of categories] array (due to one-hot encoding) as an input