Я пытаюсь получить вывод из слоя предсказаний 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').
Что-то я не так делаю?