test data used during training

[philippemiron]

Looking at the code in data.py:

idx = np.random.choice(train_data.shape[0], 1000, replace=False)
test_data = train_data[idx, :]
train_data = PinnDataset(train_data)
test_data = PinnDataset(test_data)

the testing data is used during training, this should be removed from the training dataset for appropriate testing of the model.