“class_weights” problem

[hehuan710]

keras-cam/cam.py

Line 34 in 2b7ada2

for i, w in enumerate(class_weights[:, 1]):

Should this be the case here?
for i, w in enumerate(class_weights[:,np.argmax(predictions)]):