def get_swiss_roll_data(n_samples=1000):
    noise = 0.2
    (X, _) = make_swiss_roll(n_samples, noise)
    X = X.astype('float32')[:, [0, 2]]
    return (X, _)
def get_swiss_roll_data(n_samples=1000):
    noise = 0.2
    (X, _) = make_swiss_roll(n_samples, noise)
    X = X.astype('float32')[:, [0, 2]]
    return (X, _)