linkedin / detext

DeText: A Deep Neural Text Understanding Framework for Ranking and Classification Tasks
BSD 2-Clause "Simplified" License
1.26k stars 133 forks source link

[detext] Adding multilabel classification #76

Closed amberkaur1 closed 1 year ago

amberkaur1 commented 2 years ago

Description

Added multi-label classification task to detext. Updated input label format to array (shape [num_classes], assuming one-hot array input), changed loss and activation functions, added precision/recall metrics. Passed unit tests on detect-dev mp while being backwards compatible.

Please delete options that are not relevant.

List all changes

Please list all changes in the commit.

Testing

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration

Passed unit tests on detect-dev mp while being backwards compatible. Also passed unit tests on li-detext, but full pipeline with real data yet to be confirmed

Test Configuration:

Checklist

amberkaur1 commented 1 year ago

Thank you for your effort on multilabel classification! Please check my comments below.

Updated the code accordingly, thanks for reviewing!