Skip to content

Commit

Permalink
Fix invalid type
Browse files Browse the repository at this point in the history
  • Loading branch information
keisen committed Oct 6, 2023
1 parent de02308 commit 91db4d8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tf_keras_vis/scorecam.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ def __call__(self,
for prediction in listify(preds))

# Calculating weights
weights = ([score(K.softmax(p)) for p in prediction]
weights = ([score(K.softmax(tf.constant(p))) for p in prediction]
for score, prediction in zip(scores, preds))
weights = ([self._validate_weight(s, nsamples) for s in w] for w in weights)
weights = (np.array(w, dtype=np.float32) for w in weights)
Expand Down

0 comments on commit 91db4d8

Please sign in to comment.