from sklearn.datasets import load_digits from sklearn.linear_model import LogisticRegression d = load_digits(10) train_X = d.data[:1500] train_y = d.target[:1500] test_X = d.data[1500:] test_y = d.target[1500:] lr = LogisticRegression() lr.fit(train_X, train_y) pred = lr.predict(test_X) print(pred)
