Hi all,
I stuck in music_inference_model function, in step 2D.
I didn’t find how to use tf.math.argmax and tf.one_hot .
Can you help me?
Thanks before.
Hi all,
I stuck in music_inference_model function, in step 2D.
I didn’t find how to use tf.math.argmax and tf.one_hot .
Can you help me?
Thanks before.
Hi @cauwdri,
To get the index of the predicted output with max value, you need to pass ‘out’ as the parameter for tf.math.argmax()
The parameters pass to tf.one_hot()
are the index returned from tf.math.argmax(), and the depth will the n_values
Thanks for the help, it works.