Fix train test split
Change train test split to 80 / 20 as given in the description (was 90 / 10 in the code)
"y = data[:, -1]\n",
"\n",
"# Use the first 80% of the data for training, and the last 20% for testing.\n",
- "train_n = int(floor(0.9*len(X)))\n",
+ "train_n = int(floor(0.8*len(X)))\n",
"\n",
"train_x = X[:train_n, :].contiguous().cuda()\n",
"train_y = y[:train_n].contiguous().cuda()\n",
