Моя модель CNN всегда предсказывает один класс, когда я тестирую свою модель - PullRequest
Новая
       46

Моя модель CNN всегда предсказывает один класс, когда я тестирую свою модель

0 голосов
/ 25 мая 2020

Я новичок в машинном обучении. Я попытался создать модель, которую я узнал от inte rnet, и получил точность 81% для двух видов, но когда я тестирую модель, моя модель всегда предсказывает один spe cie. Помогите мне, пожалуйста, я запутался. Спасибо

Epoch 50/50 46/46 [============================= ===] - 104 с 2 с / шаг - потеря: 0,3790 - точность: 0,8185 - val_loss: 0,2218 - val_accuracy: 0,8050 

import numpy as np
from keras.preprocessing import image

img_pred = image.load_img(r'D:\Know Wood Laptop\Anatomy-master\Test\ws3\1315_3.jpg' , target_size = (150 , 150))
#img_pred = image.load_img(r'D:\Know Wood Laptop\Anatomy-master\Test\ws1\806_1.jpg' , target_size = (150 , 150))

img_pred = image.img_to_array(img_pred)
image = img_pred/255.0
img_pred = np.expand_dims(img_pred,axis=0)

rslt = model.predict(img_pred)


print(rslt)
if rslt[0][0]==0:
    prediction = "ws1"
else:
    prediction = "ws3"

print (prediction)

[[0.]] Ws1

...