def feature_normalize(train_X): global mean, std mean = np.mean(train_X, axis=0) std = np.std(train_X, axis=0) return np.nan_to_num((train_X - mean) / std)