Add PBT to experiment creation form (#1909)

kubeflow · Jun 29, 2022 · 3f2804b · 3f2804b
1 parent 2d35224
commit 3f2804b
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
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
@@ -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
@@ -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
@@ -9,6 +9,7 @@ export enum AlgorithmsEnum {
   SOBOL = 'sobol',
   ENAS = 'enas',
   DARTS = 'darts',
+  PBT = 'pbt',
 }
 
 export enum EarlyStoppingAlgorithmsEnum {