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;