microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
https://nni.readthedocs.io
MIT License
13.88k stars 1.81k forks source link

Supporting string choices for GPTuner #5778

Open TimSchim opened 1 month ago

TimSchim commented 1 month ago

What would you like to be added: Currently the search space for GPTuner only supports numerical values for type choice. A possiblity to provide strings would be a convenient option.

Why is this needed: While it is possible to replace strings with numerical encodings, this reduces readability of code and intuitive interpretability of the hyperparameter plots.

Brief description of your proposal if any: Internal encoding to numerical values if strings are provided.

Duplicate of #1546