diff --git a/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-settings.const.ts b/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-settings.const.ts index 0ca7f3c0b90..b34b2b8c01b 100644 --- a/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-settings.const.ts +++ b/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-settings.const.ts @@ -242,6 +242,29 @@ export const DartsSettings: AlgorithmSetting[] = [ }, ]; +export const PbtSettings: AlgorithmSetting[] = [ + { + name: 'suggestion_trial_dir', + value: '/var/log/katib/checkpoints/', + type: AlgorithmSettingType.STRING, + }, + { + name: 'n_population', + value: 40, + type: AlgorithmSettingType.INTEGER, + }, + { + name: 'resample_probability', + value: null, + type: AlgorithmSettingType.FLOAT, + }, + { + name: 'truncation_threshold', + value: 0.2, + type: AlgorithmSettingType.FLOAT, + }, +]; + export const EarlyStoppingSettings: AlgorithmSetting[] = [ { name: 'min_trials_required', @@ -271,4 +294,5 @@ export const AlgorithmSettingsMap: { [key: string]: AlgorithmSetting[] } = { [AlgorithmsEnum.SOBOL]: SOBOLSettings, [AlgorithmsEnum.ENAS]: ENASSettings, [AlgorithmsEnum.DARTS]: DartsSettings, + [AlgorithmsEnum.PBT]: PbtSettings, }; diff --git a/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-types.const.ts b/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-types.const.ts index 607e6f2db63..4f1f0aa0cfd 100644 --- a/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-types.const.ts +++ b/pkg/new-ui/v1beta1/frontend/src/app/constants/algorithms-types.const.ts @@ -12,6 +12,7 @@ export const AlgorithmNames = { [AlgorithmsEnum.MULTIVARIATE_TPE]: 'Multivariate Tree of Parzen Estimators', [AlgorithmsEnum.CMAES]: 'Covariance Matrix Adaptation: Evolution Strategy', [AlgorithmsEnum.SOBOL]: 'Sobol Quasirandom Sequence', + [AlgorithmsEnum.PBT]: 'Population Based Training', }; export const NasAlgorithmNames = { diff --git a/pkg/new-ui/v1beta1/frontend/src/app/enumerations/algorithms.enum.ts b/pkg/new-ui/v1beta1/frontend/src/app/enumerations/algorithms.enum.ts index 425d423199d..8f7476468b1 100644 --- a/pkg/new-ui/v1beta1/frontend/src/app/enumerations/algorithms.enum.ts +++ b/pkg/new-ui/v1beta1/frontend/src/app/enumerations/algorithms.enum.ts @@ -9,6 +9,7 @@ export enum AlgorithmsEnum { SOBOL = 'sobol', ENAS = 'enas', DARTS = 'darts', + PBT = 'pbt', } export enum EarlyStoppingAlgorithmsEnum {