Пользовательская функция потерь: логиты и цели должны иметь одинаковую форму ((?, 1) против (45000,))

Моя модель предсказывает все как 0 в двоичном классификаторе. Всего у нас 4000 истинных и 41000 ложных. Поэтому мы пытаемся создать пользовательскую функцию потерь.

Ошибка, которую я получаю:

(logits.get_shape(), target.get_shape())))

ValueError: логиты и цели должны иметь одинаковую форму ((?, 1) против (45000,))

Код выглядит следующим образом:

combined = tf.keras.layers.concatenate([modelRNN.output, modelCNN.output])

final_dense = tf.keras.layers.Dense(10, activation='relu')(combined) #ff kijken of dit slim is
final_dense = tf.keras.layers.Dense(1, activation='sigmoid')(final_dense)

final_model = tf.keras.Model(inputs=[modelCNN.input, modelRNN.input], outputs=final_dense)

targets = match_train
logits = final_dense
pos_weight = (45000 - 4539) / 4539


custom_loss = tf.nn.weighted_cross_entropy_with_logits(
    targets,
    logits,
    pos_weight,
    )


final_model.compile(optimizer='adam',
                    loss=custom_loss,
                    metrics=['accuracy'])

форма исходных массивов:

modelCNN = (45000, 28, 28, 1) float64
modelRNN = (45000, 93, 13) float64
labels = (45000,1) boolean

Проблема была частично решена с помощью кода в комментариях. Теперь я получаю сообщение об ошибке, которого раньше не было. В нем говорится:

TypeError: Using a `tf.Tensor` as a Python `bool` is not allowed. Use `if t is not None:` instead of `if t:` to test if a tensor is defined, and use TensorFlow ops such as tf.cond to execute subgraphs conditioned on the value of a tensor.

  File "<ipython-input-6-42327e5a4b50>", line 3, in <module>
    metrics=['accuracy'])

  File "C:\Users\Tijev\Anaconda3\envs\tfp3.6\lib\site-packages\tensorflow\python\training\checkpointable\base.py", line 442, in _method_wrapper
    method(self, *args, **kwargs)

  File "C:\Users\Tijev\Anaconda3\envs\tfp3.6\lib\site-packages\tensorflow\python\keras\engine\training.py", line 215, in compile
    loss = loss or {}