import torch
import numpy as np
import sklearn.datasets

# This dataset is just fitting a linear equation, where x is sampled uniformly and y=2x+.5 
torch.manual_seed(42);np.random.seed(42)

x,y=sklearn.datasets.make_moons(5000)

torch.save((torch.from_numpy(x),torch.from_numpy(y)),"/data/cameron/toy_1d_datasets/moons.pt")