Chapter 3 (Page 82): Getting error during Fitting the SGD Classifier with Training data

[tanvirnwu]

I getting errors for these lines of codes:

from sklearn.linear_model import SGDClassifier
y_train_5 = (y_train == 5)
y_test_5 = (y_test == 5)
sgd_clf = SGDClassifier(max_iter=5, tol=1e-3, random_state=42)
sgd_clf.fit(X_train, y_train_5)

ValueError: The number of classes has to be greater than one; got 1 class

How can I solve this issue? Please guide me. Thank you so much.