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

def round(self, value_array):
        """
        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
        """

        rounded_values = np.zeros(value_array.shape)
        rounded_values[np.argmax(value_array)] = 1
        return rounded_values