TypeError: только массивы размера 1 могут быть преобразованы в Python скаляры при использовании record_set в модели LinearLearner - PullRequest
Новая
       55

TypeError: только массивы размера 1 могут быть преобразованы в Python скаляры при использовании record_set в модели LinearLearner

0 голосов
/ 17 марта 2020

Я попытался сгенерировать record_set, используя модель LinearLearner в sagemaker. Итак, сначала я передал train_data как ndarray и пометил как 1-darray 

labels_vector=np.array(train_labels).flatten()
formatted_train_data = estimator.record_set(train=train_features,
                                            labels=labels_vector)

, но он дал мне ошибку TypeError: only size-1 arrays can be converted to Python scalars, поэтому после поиска я использовал np.vectorize, чтобы передать его как вектор 

labels_vector=np.array(train_labels).flatten()
formatted_train_data = estimator.record_set(train=train_features,
                                            labels=np.vectorize(labels_vector))

, тогда это увеличивает Err AttributeError: 'vectorize' object has no attribute 'shape'

...