Multi-target classification in tensorflow

434 views Asked by At

How to implement Multi-target classification in tensorflow? Say I have a list of features f1,f2,...,fn and i want to predict the class or a value of three targets t1,t2,and t3. So each target belongs to a single class only.

1

There are 1 answers

0
MatthewScarpino On

It sounds like you're interested in multinomial logistic regression. In TensorFlow, the most important function for this is tf.nn.softmax_cross_entropy_with_logits_v2.

This site gives a good idea how the softmax function makes it possible to classify a point in multiple categories.