Блог

Я использую данные mnist из aws

 urllib.request.urlretrieve("https://s3.amazonaws.com/img-datasets/mnist.pkl.gz", "mnist.pkl.gz")
 

но у меня есть ошибка

 ValueError: Array must be a Matrix
 

форма (60000, 28, 28) и мой учебный код позади :

 kmeans = KMeans(role=role,
     instance_count=1,
     instance_type='ml.t2.medium',
     output_path=output_location,
     k=10,
     data_location=data_location)

kmeans.fit(kmeans.record_set(train_set[0]))