yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-03 19:29:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix zerodivision (#1000)

Browse Source 
* fix zerodivision

* update

* remove final

---------

Co-authored-by: Li Jiang <lijiang1@microsoft.com>
This commit is contained in:
Susan Xueqing Liu 2023-04-22 23:55:51 -04:00 committed by GitHub
parent da0d8c05e1
commit 7114b8f742
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
flaml/automl/model.py
View File

@ -1136,8 +1136,7 @@ class TransformersEstimator(BaseEstimator):
except ZeroDivisionError:
logger.warning("Zero division error appeared in HuggingFace Transformers.")
predictions = np.array([-0.05] * len(test_dataset))
else:
return predictions
return predictions
def score(self, X_val: DataFrame, y_val: Series, **kwargs):
import transformers
@ -1169,14 +1168,13 @@ class TransformersEstimator(BaseEstimator):
kwargs = {} if self._task not in NLG_TASKS else {"metric_key_prefix": "predict"}
try:
predictions = new_trainer.predict(test_dataset, **kwargs)
predictions = new_trainer.predict(test_dataset, **kwargs).predictions
except ZeroDivisionError:
logger.warning("Zero division error appeared in HuggingFace Transformers.")
predictions = np.array([0] * len(test_dataset))
post_y_pred, _ = postprocess_prediction_and_true(
task=self._task,
y_pred=predictions.predictions,
y_pred=predictions,
tokenizer=self.tokenizer,
hf_args=self._training_args,
X=X,