source/FLAML
1
0
Fork 0
mirror of https://github.com/microsoft/FLAML.git synced 2026-02-09 02:09:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert c6a5dd1a0

Browse Source
This commit is contained in:
Mark Harley
2023-02-19 00:38:54 +00:00
parent 33b1a1853b
commit e55e35adea
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
flaml/automl/model.py
View File

@@ -571,7 +571,9 @@ class TransformersEstimator(BaseEstimator):
@property
def data_collator(self):
from flaml.automl.task.task import Task
from flaml.automl.nlp.huggingface.data_collator import task_to_datacollator_class
from flaml.automl.nlp.huggingface.data_collator import (
task_to_datacollator_class,
)
data_collator_class = task_to_datacollator_class.get(
self._task.name if isinstance(self._task, Task) else self._task
@@ -2040,7 +2042,7 @@ class TS_SKLearn(SKLearnEstimator):
space.update(
{
"optimize_for_horizon": {
"domain": tune.choice([False]),
"domain": tune.choice([True, False]),
"init_value": False,
"low_cost_init_value": False,
},