tensorflow / adanet

Fast and flexible AutoML with learning guarantees.
https://adanet.readthedocs.io
Apache License 2.0
3.47k stars 527 forks source link

Multi-label classification #141

Closed EstebanGuillen closed 4 years ago

EstebanGuillen commented 4 years ago

Would it be possible to support multi-label classification?

cweill commented 4 years ago

@EstebanGuillen: adanet already supports multi-label classification. See https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/estimator/multi_label_head?hl=sv-SESee