-
Notifications
You must be signed in to change notification settings - Fork 2.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[AutoNLP] Add english models for text classification #4704
Conversation
Thanks for your contribution! |
Codecov Report
@@ Coverage Diff @@
## develop #4704 +/- ##
========================================
Coverage 42.97% 42.98%
========================================
Files 434 434
Lines 61956 61965 +9
========================================
+ Hits 26626 26636 +10
+ Misses 35330 35329 -1
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
@@ -129,19 +133,42 @@ def _model_candidates(self) -> List[Dict[str, Any]]: | |||
"ernie-3.0-nano-zh", # 4-layer, 312-hidden, 12-heads, 18M parameters. | |||
], | |||
) | |||
english_models = hp.choice( |
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.
添加英文模型的单测
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.
英文模型和中文模型逻辑一致,所以我没有添加新的测试。不过增加了test_model_language_filter来确保选英文能拿到英文模型的candidates
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
PR types
PR changes
Description