Skip to content

Commit

Permalink
add info about HPO methods (#391)
Browse files Browse the repository at this point in the history
  • Loading branch information
pplonski committed Apr 29, 2021
1 parent e83e639 commit ede835a
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ The example for `LightGBM` summary:
![Decision Tree summary](https://github.com/mljar/mljar-examples/blob/master/media/lightgbm_summary.gif)


## Available Modes :books:
## Available Modes

In the [docs](https://supervised.mljar.com/features/modes/) you can find details about AutoML modes are presented in the table .

Expand Down Expand Up @@ -289,6 +289,11 @@ print("Test MSE:", mean_squared_error(y_test, predictions))

# FAQ

<details><summary>What method is used for hyperparameters optimization?</summary>
- For modes: `Explain`, `Perform` and `Compete` there is used a random search method combined with hill climbing. In this approach all checked models are saved and used for building Ensemble.
- For mode: `Optuna` the Optuna framework is used. It is using TPE sampler for tuning. Models checked during Optuna hyperparameters search are not saved, only the best model is saved (final model from tuning). You can check the details about checked hyperparameters from optuna by checking study files in `optuna` directory in your AutoML `results_path`.
</details>

<details><summary>How to save and load AutoML?</summary>

The save and load of AutoML models is automatic. All models created during AutoML training are saved in the directory set in `results_path` (argument of `AutoML()` constructor). If there is no `results_path` set, then the directory is created based on following name convention: `AutoML_{number}` the `number` will be number from 1 to 1000 (depends which directory name will be free).
Expand Down

0 comments on commit ede835a

Please sign in to comment.