def check_normalization_constants(arr, axis): sum = np.log(np.sum(arr, axis=axis)) z = normalize(np.log(arr), axis=axis)[0] zdiff = np.abs(sum - z) if not (zdiff < thresh).all(): print sum print z raise AssertionError("wrong normalization constant")
def check_normalization(arr, axis): sum = np.sum(arr, axis=axis) norm = np.log(arr / np.expand_dims(sum, axis=axis)) n = normalize(np.log(arr), axis=axis)[1] ndiff = np.abs(norm - n) if not (ndiff < thresh).all(): print norm print n raise AssertionError("wrong normalized values")
def check_normalization(arr, axis): sum = np.sum(arr, axis=axis) norm = np.log(arr / np.expand_dims(sum, axis=axis)) n = normalize(np.log(arr), axis=axis)[1] ndiff = np.abs(norm - n) if not(ndiff < thresh).all(): print norm print n raise AssertionError("wrong normalized values")