Невозможно импортировать ранее обученную модель scikit-learn: не удается получить атрибут MiniBatchKMeans - PullRequest
Новая
       37

Невозможно импортировать ранее обученную модель scikit-learn: не удается получить атрибут MiniBatchKMeans

0 голосов
/ 14 февраля 2020

Ранее я обучал модель MiniBatchKMeans из Scikit-learn и сохранял модель, используя: 

pickle.dump(kmeans, open("/path/to/model", "wb"))

Теперь, когда я хочу загрузить эту модель, она выдает мне эту ошибку: 

Python 3.7.2 (default, Dec 29 2018, 06:19:36)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import pickle
>>> pickle.load(open("/path/to/model", "rb"))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: Can't get attribute 'MiniBatchKMeans' on <module 'sklearn.cluster._k_means' from '/root/anaconda/lib/python3.7/site-packages/sklearn/cluster/_k_means.cpython-37m-x86_64-linux-gnu.so'>

Среда с обеих сторон (место, где я тренируюсь, и место, куда я загружаю его) должны быть одинаковыми, поскольку они из одного контейнера Docker, имеют одинаковую версию Python и версию Scikit-learn, просто запускаю другой код.

Есть мысли о том, почему это произойдет?

...