ValueError: Ошибка при проверке ввода: ожидалось, что input_1 будет иметь 4 измерения, но получил массив с формой (83, 1)

Я хочу найти вероятности для моей модели, чтобы позже найти AUC и ROC, но получил эту ошибку:

ValueError: Ошибка при проверке ввода: ожидалось, что input_1 будет иметь 4 измерения, но получил массив с формой (83, 1)

Мой код:

def create_resnet(img_dim,CHANNEL,n_class):
    input_tensor=Input(shape=(img_dim, img_dim,CHANNEL))

    base_model = ResNet50(weights=None, include_top=False, input_tensor=input_tensor)
    base_model.load_weights('D:/Diabetic retinopathy data set/ResNet50/resnet50_weights_tf_dim_ordering_tf_kernels_notop.h5')

    x=GlobalAveragePooling2D()(base_model.output)
    x=Dropout(0.3)(x)
    x=Dense(1024, activation=relu)(x)
    x=Dropout(0.2)(x)
    x=Dense(512, activation=relu)(x)
    x=Dropout(0.2)(x)
    x=BatchNormalization()(x)
    output_layer=Dense(n_class,activation='softmax', name="Output_Layer")(x)
    model_resnet =Model(input_tensor, output_layer)

    return model_resnet

model_resnet=create_resnet(IMG_DIM,CHANNEL_SIZE, NUM_CLASSES)