Выход VGG из слоя предсказаний в nan после применения функций суммирования по времени - PullRequest
0 голосов
/ 27 апреля 2019

Я пытаюсь получить вывод из слоя предсказаний VGG, используя следующее:

   input_shape= (224, 224, 3)
   vgg = vgg16.VGG16(include_top=True, weights='imagenet', 
   input_shape=input_shape)

   model_output = vgg.get_layer("predictions").output
   vgg_model = Model(inputs=vgg.input, outputs=model_output)

Затем, после добавления вывода для каждого изображения в папке, я применяю функции суммирования

    func_list = [np.average,  np.max,  np.std , np.min, np.mean, 
    scipy.stats.skew,scipy.stats.kurtosis]

    suffix_list = ['_avg', '_max', '_std','_min' ,'_mean' ,'_skew','_kurtosis']

и затем я хочу проверить вывод с использованием классификатора случайного леса:

 yc = y_train
 val_yc =y_test

 train_feat= vgg_train
 test_feat= vgg_test


max_features = list(range(1,train_feat.shape[1]))

для max_feature в max_features:

sel = SelectFromModel(RandomForestClassifier(n_estimators=2000),max_features=max_feature)
sel.fit(train_feat, yc)

train_c = train_feat.loc[:, sel_c.get_support()]
test_c  = test_feat.loc[:, sel_c.get_support()]

Я получаю эту ошибку каждый раз:

  ValueError: Input contains NaN, infinity or a value too large for dtype('float32').

Что-то я не так делаю?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...