diff --git a/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx b/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx index 65ad4c8b..fded3102 100644 --- a/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx +++ b/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx @@ -21,6 +21,7 @@ const classificationAlgorithms = [ 'lda', 'liblinear_svc', 'libsvm_svc', + 'tablenet', 'mlp', 'multinomial_nb', 'passive_aggressive', @@ -30,13 +31,7 @@ const classificationAlgorithms = [ 'LightGBMClassification', 'XGBoostClassification', 'StackingClassification', -].map((name) => { - if (name === 'mlp') { - return { label: 'tablenet', value: name }; - } else { - return { label: name, value: name }; - } -}); +].map((name) => ({ label: name, value: name })); // 回归算法 const regressorAlgorithms = [