diff --git a/react-ui/src/pages/AutoML/Instance/index.tsx b/react-ui/src/pages/AutoML/Instance/index.tsx index 09a55023..d53ecc17 100644 --- a/react-ui/src/pages/AutoML/Instance/index.tsx +++ b/react-ui/src/pages/AutoML/Instance/index.tsx @@ -12,7 +12,7 @@ import AutoMLBasic from '../components/AutoMLBasic'; import ExperimentHistory from '../components/ExperimentHistory'; import ExperimentLog from '../components/ExperimentLog'; import ExperimentResult from '../components/ExperimentResult'; -import TensorBoard from '../components/TensorBoard'; +import ExperimentVisualResult from '../components/ExperimentVisualResult'; import { AutoMLData, AutoMLInstanceData } from '../types'; import styles from './index.less'; @@ -186,13 +186,13 @@ function AutoMLInstance() { /> ), }, - type === AutoMLType.Text + type === AutoMLType.Text || type === AutoMLType.Video ? { key: TabKeys.Visual, label: '可视化结果', icon: , children: ( - diff --git a/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx b/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx index 3f85f5c5..c548e079 100644 --- a/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx +++ b/react-ui/src/pages/AutoML/components/CreateForm/ExecuteConfig.tsx @@ -27,6 +27,9 @@ const classificationAlgorithms = [ 'qda', 'random_forest', 'sgd', + 'LightGBMClassification', + 'XGBoostClassification', + 'StackingClassification', ].map((name) => ({ label: name, value: name })); // 回归算法 @@ -43,6 +46,8 @@ const regressorAlgorithms = [ 'mlp', 'random_forest', 'sgd', + 'LightGBMRegression', + 'XGBoostRegression', ].map((name) => ({ label: name, value: name })); // 特征预处理算法 @@ -289,7 +294,7 @@ function ExecuteConfig() { name="ensemble_nbest" tooltip="仅集成最佳的N个模型" > - + @@ -421,7 +426,7 @@ function ExecuteConfig() { }, ]} > - + diff --git a/react-ui/src/pages/AutoML/components/ExperimentResult/index.tsx b/react-ui/src/pages/AutoML/components/ExperimentResult/index.tsx index 8900b2a3..375b416d 100644 --- a/react-ui/src/pages/AutoML/components/ExperimentResult/index.tsx +++ b/react-ui/src/pages/AutoML/components/ExperimentResult/index.tsx @@ -1,5 +1,6 @@ import InfoGroup from '@/components/InfoGroup'; import { AutoMLType } from '@/enums'; +import TrialFileTree from '@/pages/HyperParameter/components/TrialFileTree'; import { getFileReq } from '@/services/file'; import { to } from '@/utils/promise'; import { Button, Image } from 'antd'; @@ -40,7 +41,11 @@ function ExperimentResult({ fileUrl, imageUrl, modelPath, type }: ExperimentResu return (
-
{result}
+ {type === AutoMLType.Video ? ( + + ) : ( +
{result}
+ )}
{type === AutoMLType.Table && ( diff --git a/react-ui/src/pages/AutoML/components/TensorBoard/index.tsx b/react-ui/src/pages/AutoML/components/ExperimentVisualResult/index.tsx similarity index 90% rename from react-ui/src/pages/AutoML/components/TensorBoard/index.tsx rename to react-ui/src/pages/AutoML/components/ExperimentVisualResult/index.tsx index e05a9452..1837d625 100644 --- a/react-ui/src/pages/AutoML/components/TensorBoard/index.tsx +++ b/react-ui/src/pages/AutoML/components/ExperimentVisualResult/index.tsx @@ -15,7 +15,7 @@ type TensorBoardProps = { path?: string; }; -function TensorBoard({ namespace, path }: TensorBoardProps) { +function ExperimentVisualResult({ namespace, path }: TensorBoardProps) { const [tensorboardUrl, setTensorboardUrl] = useState(''); useEffect(() => { // 运行 TensorBoard @@ -40,4 +40,4 @@ function TensorBoard({ namespace, path }: TensorBoardProps) { return <>{tensorboardUrl && }; } -export default TensorBoard; +export default ExperimentVisualResult;