()
| 258 | |
| 259 | |
| 260 | def get_data(): |
| 261 | ### medical data |
| 262 | # load the data |
| 263 | # data = load_breast_cancer() |
| 264 | # X, Y = data.data, data.target |
| 265 | |
| 266 | # X, Y = get_xor() |
| 267 | # X, Y = get_donut() |
| 268 | # X, Y = get_spiral() |
| 269 | X, Y = get_clouds() |
| 270 | |
| 271 | # split the data into train and test sets |
| 272 | # this lets us simulate how our model will perform in the future |
| 273 | Xtrain, Xtest, Ytrain, Ytest = train_test_split(X, Y, test_size=0.33) |
| 274 | return Xtrain, Xtest, Ytrain, Ytest |
| 275 | |
| 276 | |
| 277 | if __name__ == '__main__': |