yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-01-08 05:01:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

update model of text summarization (#1030)

Browse Source
This commit is contained in:
Susan Xueqing Liu 2023-05-09 20:48:22 -04:00 committed by GitHub
parent 59e882e5cc
commit f01acb67f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/nlp/test_autohf_summarization.py
View File

@ -22,7 +22,7 @@ def test_summarization():
automl_settings["task"] = "summarization"
automl_settings["metric"] = "rouge1"
automl_settings["time_budget"] = 2 * automl_settings["time_budget"]
automl_settings["fit_kwargs_by_estimator"]["transformer"]["model_path"] = "patrickvonplaten/t5-tiny-random"
automl_settings["fit_kwargs_by_estimator"]["transformer"]["model_path"] = "google/flan-t5-small"
try:
automl.fit(X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings)