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