Submitted by ole72444 t3_yuafo2 in deeplearning
FuB4R32 t1_iw8kirz wrote
Reply to comment by sckuzzle in Making a model predict on the basis of a particular value by ole72444
You could also do this at the input if its hard to edit the training data, e.g. in tensorflow https://www.tensorflow.org/api_docs/python/tf/gather
https://www.tensorflow.org/api_docs/python/tf/math/argmax
Generally, should look into custom operations like this to achieve what you want
Viewing a single comment thread. View all comments