kubeflow / sdk / 18977374914

66%

push

github

feat: Hyperparameter Optimization APIs in Kubeflow SDK (#124)

* Init commit

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Create optimize() API

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Set retain=True for Experiment

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Fix location to Trainer utils

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Implement get_job, list_jobs, and delete_job APIs

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Add metrics and parameters to Trial object

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Clarify message for objective

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Move TrainJobTemplate to the Trainer types

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Rename CRD to CR

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Fix serialization of TrainJob

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Rename ExecutionBackend to RuntimeBackend

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Export GridSearch
Rename common types

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Add OptimizationJob constant

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Change to BaseAlgorithm

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Keep func_args for Trainer

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Use PyPI package for Katib models

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

* Update lock file

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

---------

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>

32 of 38 new or added lines in 1 file covered. (84.21%)

168 of 211 relevant lines covered (79.62%)

1.59 hits per line