from scipy.io import loadmat
mnist = loadmat('datasets/MNIST/mnist-original.mat')
mnist["target"]=mnist["label"]
del mnist["label"]
mnist
X,y = mnist["data"],mnist["target"]
X = X.T
X.shape
y=y.T.ravel()
y.shape
X_train,X_test,y_train,y_test = X[:60000],X[60000:],y[:60000],y[60000:]
import numpy as np
shuffle_index = np.random.permutation(60000)
X_train,y_train = X_train[shuffle_index],y_train[shuffle_index]
y_train_5 = (y_train == 5)
y_test_5 = (y_test == 5)
from sklearn.linear_model import SGDClassifier
sgd_clf = SGDClassifier(random_state=42)
sgd_clf.fit(X_train,y_train_5)
sgd_clf.predict([some_digit