Problems with mackey example

Here is what i’ve found: 1.

if dtype == torch.double:
    data = np.loadtxt('../../datasets/mg17.csv', delimiter=',', dtype=np.float64)
elif dtype == torch.float:
data = np.loadtxt('../../datasets/mg17.csv', delimiter=',', dtype=np.float32)

Here are wrong pathes to dataset since it’s in the same directory so it should be

if dtype == torch.double:
    data = np.loadtxt('datasets/mg17.csv', delimiter=',', dtype=np.float64)
elif dtype == torch.float:
    data = np.loadtxt('datasets/mg17.csv', delimiter=',', dtype=np.float32)

1. Next.

X_data = np.expand_dims(data[:, [0]], axis=1)
Y_data = np.expand_dims(data[:, [1]], axis=1)

I guess it should be

X_data = np.expand_dims(data[:, [1]], axis=1)
Y_data = np.expand_dims(data[:, [0]], axis=1)

or am i wrong? 3. When launching with these fixes i’m getting

Training error: 0.051316608186617714 Test error: 0.052949286276803606

It’s quite big. Moreover, when i’m trying to make plot to compare real and output i’m getting following picture on “test” data

5000 test outputs

Closer look (100 outputs)

Am i doing something wrong?

[stefanonardo]

Ok, I see your point now! The dataset is the problem, second column is wrong. I don’t how it happened. I will fix the repo ASAP, in the meantime, try this file: https://gist.github.com/stefanonardo/e216b13b21d07d31ca557058028c1198

[DaddyWesker]

Ok, I see your point now! The dataset is the problem, second column is wrong. I don’t how it happened. I will fix the repo ASAP, in the meantime, try this file: https://gist.github.com/stefanonardo/e216b13b21d07d31ca557058028c1198

Great. That worked. Thanks