np
.
argmax
(
keras_cat_classes
,
axis
=
1
)
Hosted on
Deepnote