Изменение формы массивов, чтобы они могли соответствовать модели машинного обучения - PullRequest
0 голосов
/ 12 октября 2019

Я делаю классификатор изображений, но у меня проблема с изменением формы данных. Мой набор данных (на данный момент) очень маленький, он содержит 19 изображений, размером 75x75.

Когда я обработал свои фотографии, я получил два массива для функций и результатов:

Форма для функций:(19, 75, 75)

Форма для результатов: (19,)

Когда я передаю эти массивы в мою модель, я получаю эту ошибку:

clf = RandomForestClassifier(n_estimators=100, random_state=0)
model = clf.fit(features, results)

ValueError: Found array with dim 3. Estimator expected <= 2.

Я не знаю, какчтобы изменить эти данные, не могли бы вы помочь мне? Я хочу добавить свое изображение, чтобы я мог классифицировать его:

my_image = r'C:\Users\Pc\Desktop\image_classifier\Prediction\predikcija.jpg'
img_array = cv2.imread(my_image, 0)
img_array = cv2.resize(img_array, (img_size,img_size))

my_prediction = model.predict(img_array)
print(my_prediction)

Я получаю эту ошибку:

   ValueError: Number of features of the model must match the input. Model n_features is 6400 and input n_features is 80  
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...