Skip to content

Commit

Permalink
✨Enhanced Auto
Browse files Browse the repository at this point in the history
Now we support specifying `extra_config` with json file path
  • Loading branch information
carefree0910 committed Dec 22, 2020
1 parent 8de370e commit 752f419
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion cflearn/api/auto.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,8 @@
from .production import Pack
from .production import Predictor
from ..types import data_type
from ..types import general_config_type
from ..configs import _parse_config
from ..protocol import DataProtocol


Expand Down Expand Up @@ -135,11 +137,12 @@ def fit(
estimator_scoring_function: Union[str, scoring_fn_type] = default_scoring,
temp_folder: str = "__tmp__",
num_final_repeat: int = 20,
extra_config: Optional[Dict[str, Any]] = None,
extra_config: general_config_type = None,
cuda: Optional[Union[str, int]] = None,
) -> "Auto":
self.best_params = {}
self.optuna_results = {}
extra_config = _parse_config(extra_config)

tuner = None
optuna_folder = os.path.join(temp_folder, "__optuna__")
Expand Down

0 comments on commit 752f419

Please sign in to comment.