add convert_to_tensor for perplexity

keras-team · Aug 17, 2023 · 983b7a6 · 983b7a6
1 parent e4531ad
commit 983b7a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_nlp/metrics/perplexity.py b/keras_nlp/metrics/perplexity.py
@@ -161,7 +161,7 @@ def update_state(self, y_true, y_pred, sample_weight=None):
 
     def result(self):
         perplexity_score = ops.where(
-            ops.equal(self._number_of_samples, 0),
+            ops.equal(ops.convert_to_tensor(self._number_of_samples), 0),
             0,
             ops.exp(self._aggregate_crossentropy / self._number_of_samples),
         )