KeyError: 85

from sklearn.datasets import fetch_openml
mnist = fetch_openml ('mnist_784')

from sklearn import preprocessing
import numpy as np
X = mnist.data.astype(float)
X = X/255
 
y = preprocessing.LabelBinarizer().fit_transform(mnist.target)
 
train_size = 60000
 
X_train, X_test = np.split(X, [train_size])
y_train, y_test = np.split(y, [train_size])

import matplotlib.pyplot as plt
 
I = 85 # номер изображения для вывода
plt.imshow(X_train[I].reshape([28, 28]), cmap='gray')
plt.title('Class: '+ str(y_train[I]))
print("Цифра: ", np.argmax(y_train[I]))

KeyError                                  Traceback (most recent call last)
/usr/local/lib/python3.7/dist-packages/pandas/core/indexes/base.py in get_loc(self, key, method, tolerance)
   3360             try:
-> 3361                 return self._engine.get_loc(casted_key)
   3362             except KeyError as err:
 
4 frames
/usr/local/lib/python3.7/dist-packages/pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()
 
/usr/local/lib/python3.7/dist-packages/pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()
 
pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
 
pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()
 
KeyError: 85
 
The above exception was the direct cause of the following exception:
 
KeyError                                  Traceback (most recent call last)
<ipython-input-41-efcd0390404f> in <module>
      2 
      3 I = 85 # номер изображения для вывода
----> 4 plt.imshow(X_train[I].reshape([28, 28]), cmap='gray')
      5 plt.title('Class: '+ str(y_train[I]))
      6 print("Цифра: ", np.argmax(y_train[I]))
 
/usr/local/lib/python3.7/dist-packages/pandas/core/frame.py in __getitem__(self, key)
   3456             if self.columns.nlevels > 1:
   3457                 return self._getitem_multilevel(key)
-> 3458             indexer = self.columns.get_loc(key)
   3459             if is_integer(indexer):
   3460                 indexer = [indexer]
 
/usr/local/lib/python3.7/dist-packages/pandas/core/indexes/base.py in get_loc(self, key, method, tolerance)
   3361                 return self._engine.get_loc(casted_key)
   3362             except KeyError as err:
-> 3363                 raise KeyError(key) from err
   3364 
   3365         if is_scalar(key) and isna(key) and not self.hasnans:
 
KeyError: 85

from google.colab.patches import cv2_imshow
cv2_imshow(img)

 %matplotlib widget

from sklearn.datasets import fetch_openml
mnist = fetch_openml ('mnist_784', as_frame=False)