Browse Source

update model of text summarization (#1030)

tags/v0.1.0rc1
Susan Xueqing Liu GitHub 3 years ago
parent
commit
f01acb67f6
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test/nlp/test_autohf_summarization.py

+ 1
- 1
test/nlp/test_autohf_summarization.py View File

@@ -22,7 +22,7 @@ def test_summarization():
automl_settings["task"] = "summarization" automl_settings["task"] = "summarization"
automl_settings["metric"] = "rouge1" automl_settings["metric"] = "rouge1"
automl_settings["time_budget"] = 2 * automl_settings["time_budget"] 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: try:
automl.fit(X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings) automl.fit(X_train=X_train, y_train=y_train, X_val=X_val, y_val=y_val, **automl_settings)


Loading…
Cancel
Save