import numpy as np from learn import Learn data = np.array([[1, 2], [2, 4], [3, 6], [4, 8], [5, 10], [6, 12], [7, 14], [8, 16], [9, 18], [10, 20], [23, 46], [54, 108], [77, 154], [18, 36], [25, 50], [16, 32], [63, 126], [41, 82], [33, 66], [45, 90], [88, 176], [64, 128], [54, 108], [98, 196], [14, 28], [34, 68]]) X, y = data[:, :-1], data[:, -1] test = [85, 17, 457, 991, 13, 68, 19, 85, 36, 1568, 997] Learn.play(X, y, test)