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