multilabel-classification

Clasificación de Texto Multilabel usando TensorFlow

Los datos de texto se organizan como vectores con 20.000 elementos, como [2, 1, 0, 0, 5, ...., 0]. el elemento i-ésimo indic ... tf.argmax(out, 1), tf.argmax(y, 1)) accuracy = tf.reduce_mean(tf.cast(correct_pred, "float")) return loss, accuracy