from sklearn.datasets import load_digits from sklearn.linear_model import LogisticRegression d = load_digits(10) x1 = d.data[:1500] y1 = d.target[:1500] x2 = d.data[1500:] y2 = d.target[1500:] lr = LogisticRegression() lr.fit(x1, y1) pred = lr.predict(x2) print(pred)