-
Notifications
You must be signed in to change notification settings - Fork 211
Add from_lists
to TextClassificationData
#805
Add from_lists
to TextClassificationData
#805
Conversation
Codecov Report
@@ Coverage Diff @@
## master #805 +/- ##
==========================================
+ Coverage 88.97% 89.05% +0.08%
==========================================
Files 219 219
Lines 11879 11916 +37
==========================================
+ Hits 10569 10612 +43
+ Misses 1310 1304 -6
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
flash/text/classification/data.py
Outdated
hf_dataset = Dataset.from_dict({"input": input, "labels": target}) | ||
|
||
if not self.predicting: | ||
dataset.multi_label = False |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Would you mind adding support for multi label ?
from_list
to TextClassificationData
from_lists
to TextClassificationData
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM ! Great work !
Thank you @kingyiusuen!! |
What does this PR do?
Fixes #769 (issue)
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Did you have fun?
Make sure you had fun coding 🙃