import numpy as np from adaboost import AdaBoost X=np.array([i for i in range(0,10)]).reshape(1,10) Y=[1,1,1,-1,-1,-1,1,1,1,-1] ada = AdaBoost(X,Y) ada.train(4) print ada.pred(X) == np.array(Y)