yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-29 08:49:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

updating the data collator for seq-regression to handle the dim mismatch problem (#751)

Browse Source
This commit is contained in:
Xueqing Liu 2022-10-09 00:59:31 -04:00 committed by GitHub
parent 860cbc233e
commit c01e65bb48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flaml/nlp/huggingface/data_collator.py
View File

@ -11,6 +11,7 @@ from flaml.data import (
MULTICHOICECLASSIFICATION,
SUMMARIZATION,
SEQCLASSIFICATION,
SEQREGRESSION
)
@ -51,5 +52,6 @@ task_to_datacollator_class = OrderedDict(
(MULTICHOICECLASSIFICATION, DataCollatorForMultipleChoiceClassification),
(SUMMARIZATION, DataCollatorForSeq2Seq),
(SEQCLASSIFICATION, DataCollatorWithPadding),
(SEQREGRESSION, DataCollatorWithPadding),
]
)