From 8a7c584f0219536f0a2b66c507e9b2b716295fc1 Mon Sep 17 00:00:00 2001
From: zhaowei <cp3hnu@gmail.com>
Date: Sun, 27 Apr 2025 11:48:19 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=88=9B=E5=BB=BA=E8=A7=86=E9=A2=91?=
 =?UTF-8?q?=E5=88=86=E7=B1=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../components/CreateForm/ExecuteConfig.tsx   |   4 +-
 react-ui/src/pages/AutoML/Create/index.tsx    |   7 +-
 .../AutoML/components/AutoMLBasic/index.tsx   |  49 +++-
 .../CreateForm/TextExecuteConfig.tsx          |   4 +-
 .../CreateForm/VideoExecuteConfig.tsx         | 260 ++++++++++++++++++
 react-ui/src/pages/AutoML/types.ts            |  14 +
 6 files changed, 332 insertions(+), 6 deletions(-)
 create mode 100644 react-ui/src/pages/AutoML/components/CreateForm/VideoExecuteConfig.tsx

diff --git a/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx b/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx
index bedbe0f2..addb402c 100644
--- a/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx
+++ b/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx
@@ -251,11 +251,11 @@ function ExecuteConfig() {
                       rules={[
                         {
                           required: true,
-                          message: '请输入epochs',
+                          message: '请输入 epochs',
                         },
                       ]}
                     >
-                      <InputNumber placeholder="请输入epochs" min={0} precision={0} />
+                      <InputNumber placeholder="请输入 epochs" min={0} precision={0} />
                     </Form.Item>
                   </Col>
                 </Row>
diff --git a/react-ui/src/pages/AutoML/Create/index.tsx b/react-ui/src/pages/AutoML/Create/index.tsx
index 1489d999..53cdf8bb 100644
--- a/react-ui/src/pages/AutoML/Create/index.tsx
+++ b/react-ui/src/pages/AutoML/Create/index.tsx
@@ -18,6 +18,7 @@ import DatasetConfig from '../components/CreateForm/DatasetConfig';
 import ExecuteConfig from '../components/CreateForm/ExecuteConfig';
 import TextExecuteConfig from '../components/CreateForm/TextExecuteConfig';
 import TrialConfig from '../components/CreateForm/TrialConfig';
+import VideoExecuteConfig from '../components/CreateForm/VideoExecuteConfig';
 import { AutoMLData, FormData } from '../types';
 import styles from './index.less';
 
@@ -200,6 +201,8 @@ function CreateAutoML() {
               test_size: 0.25,
               train_size: 0.67,
               seed: 1,
+              is_validate: false,
+              is_test: false,
             }}
           >
             <BasicConfig />
@@ -212,7 +215,9 @@ function CreateAutoML() {
               </>
             ) : type === AutoMLType.Text ? (
               <TextExecuteConfig />
-            ) : null}
+            ) : (
+              <VideoExecuteConfig />
+            )}
 
             <Form.Item wrapperCol={{ offset: 0, span: 16 }} style={{ marginTop: '40px' }}>
               <Button type="primary" htmlType="submit">
diff --git a/react-ui/src/pages/AutoML/components/AutoMLBasic/index.tsx b/react-ui/src/pages/AutoML/components/AutoMLBasic/index.tsx
index d92610e5..19c4fee0 100644
--- a/react-ui/src/pages/AutoML/components/AutoMLBasic/index.tsx
+++ b/react-ui/src/pages/AutoML/components/AutoMLBasic/index.tsx
@@ -206,7 +206,54 @@ function AutoMLBasic({ info, className, runStatus, isInstance = false }: AutoMLB
         },
       ];
     } else {
-      return [];
+      return [
+        {
+          label: '数据集',
+          value: info.dataset,
+          format: formatDataset,
+        },
+        {
+          label: '类别数量',
+          value: info.num_classes,
+        },
+        {
+          label: 'batch_size',
+          value: info.batch_size,
+        },
+        {
+          label: 'epochs',
+          value: info.epochs,
+        },
+        {
+          label: '学习率',
+          value: info.lr,
+        },
+        {
+          label: '是否验证',
+          value: info.is_validate,
+          format: formatBoolean,
+        },
+        {
+          label: '训练集路径',
+          value: info.train_data_prefix,
+        },
+        {
+          label: '训练集标注文件',
+          value: info.train_file_path,
+        },
+        ...(info.is_validate
+          ? [
+              {
+                label: '验证集路径',
+                value: info.valid_data_prefix,
+              },
+              {
+                label: '验证集标注文件',
+                value: info.valid_file_path,
+              },
+            ]
+          : []),
+      ];
     }
   }, [info, getResourceDescription]);
 
diff --git a/react-ui/src/pages/AutoML/components/CreateForm/TextExecuteConfig.tsx b/react-ui/src/pages/AutoML/components/CreateForm/TextExecuteConfig.tsx
index 87a8c697..8910c473 100644
--- a/react-ui/src/pages/AutoML/components/CreateForm/TextExecuteConfig.tsx
+++ b/react-ui/src/pages/AutoML/components/CreateForm/TextExecuteConfig.tsx
@@ -105,11 +105,11 @@ function TextExecuteConfig() {
             rules={[
               {
                 required: true,
-                message: '请输入epochs',
+                message: '请输入 epochs',
               },
             ]}
           >
-            <InputNumber placeholder="请输入epochs" min={0} precision={0} />
+            <InputNumber placeholder="请输入 epochs" min={0} precision={0} />
           </Form.Item>
         </Col>
       </Row>
diff --git a/react-ui/src/pages/AutoML/components/CreateForm/VideoExecuteConfig.tsx b/react-ui/src/pages/AutoML/components/CreateForm/VideoExecuteConfig.tsx
new file mode 100644
index 00000000..425de115
--- /dev/null
+++ b/react-ui/src/pages/AutoML/components/CreateForm/VideoExecuteConfig.tsx
@@ -0,0 +1,260 @@
+import ResourceSelect, {
+  ResourceSelectorType,
+  requiredValidator,
+} from '@/components/ResourceSelect';
+import SubAreaTitle from '@/components/SubAreaTitle';
+import { Col, Form, Input, InputNumber, Row, Switch } from 'antd';
+
+function VideoExecuteConfig() {
+  return (
+    <>
+      <SubAreaTitle
+        title="执行配置"
+        image={require('@/assets/img/model-deployment.png')}
+        style={{ marginTop: '20px', marginBottom: '24px' }}
+      ></SubAreaTitle>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="数据集"
+            name="dataset"
+            rules={[
+              {
+                validator: requiredValidator,
+                message: '请选择数据集',
+              },
+            ]}
+            required
+          >
+            <ResourceSelect
+              type={ResourceSelectorType.Dataset}
+              placeholder="请选择数据集"
+              canInput={false}
+            />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="类别数量"
+            name="num_classes"
+            rules={[
+              {
+                required: true,
+                message: '请输入类别数量',
+              },
+            ]}
+          >
+            <InputNumber placeholder="请输入类别数量" min={0} precision={0} />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="batch_size"
+            name="batch_size"
+            rules={[
+              {
+                required: true,
+                message: '请输入 batch_size',
+              },
+            ]}
+          >
+            <InputNumber placeholder="请输入 batch_size" min={0} precision={0} />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="epochs"
+            name="epochs"
+            rules={[
+              {
+                required: true,
+                message: '请输入 epochs',
+              },
+            ]}
+          >
+            <InputNumber placeholder="请输入 epochs" min={0} precision={0} />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="学习率"
+            name="lr"
+            rules={[
+              {
+                required: true,
+                message: '请输入学习率',
+              },
+            ]}
+          >
+            <InputNumber placeholder="请输入学习率" min={0} />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item label="是否验证" name="is_validate" valuePropName="checked">
+            <Switch />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      {/* <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item label="是否测试" name="is_test" valuePropName="checked">
+            <Switch />
+          </Form.Item>
+        </Col>
+      </Row> */}
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="训练集路径"
+            name="train_data_prefix"
+            rules={[
+              {
+                required: true,
+                message: '请输入训练集路径',
+              },
+            ]}
+          >
+            <Input placeholder="请输入训练集路径" maxLength={64} showCount allowClear />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Row gutter={8}>
+        <Col span={10}>
+          <Form.Item
+            label="训练集标注文件"
+            name="train_file_path"
+            rules={[
+              {
+                required: true,
+                message: '请输入训练集标注文件',
+              },
+            ]}
+          >
+            <Input placeholder="请输入训练集标注文件" maxLength={64} showCount allowClear />
+          </Form.Item>
+        </Col>
+      </Row>
+
+      <Form.Item dependencies={['is_validate']} noStyle>
+        {({ getFieldValue }) => {
+          const is_validate = getFieldValue('is_validate');
+          if (is_validate) {
+            return (
+              <>
+                <Row gutter={8}>
+                  <Col span={10}>
+                    <Form.Item
+                      label="验证集路径"
+                      name="valid_data_prefix"
+                      rules={[
+                        {
+                          required: true,
+                          message: '请输入验证集路径',
+                        },
+                      ]}
+                    >
+                      <Input placeholder="请输入验证集路径" maxLength={64} showCount allowClear />
+                    </Form.Item>
+                  </Col>
+                </Row>
+
+                <Row gutter={8}>
+                  <Col span={10}>
+                    <Form.Item
+                      label="验证集标注文件"
+                      name="valid_file_path"
+                      rules={[
+                        {
+                          required: true,
+                          message: '请输入验证集标注文件',
+                        },
+                      ]}
+                    >
+                      <Input
+                        placeholder="请输入验证集标注文件"
+                        maxLength={64}
+                        showCount
+                        allowClear
+                      />
+                    </Form.Item>
+                  </Col>
+                </Row>
+              </>
+            );
+          }
+        }}
+      </Form.Item>
+
+      {/* <Form.Item dependencies={['is_test']} noStyle>
+        {({ getFieldValue }) => {
+          const is_test = getFieldValue('is_test');
+          if (is_test) {
+            return (
+              <>
+                <Row gutter={8}>
+                  <Col span={10}>
+                    <Form.Item
+                      label="测试集路径"
+                      name="test_data_prefix"
+                      rules={[
+                        {
+                          required: true,
+                          message: '请输入测试集路径',
+                        },
+                      ]}
+                    >
+                      <Input placeholder="请输入测试集路径" maxLength={64} showCount allowClear />
+                    </Form.Item>
+                  </Col>
+                </Row>
+
+                <Row gutter={8}>
+                  <Col span={10}>
+                    <Form.Item
+                      label="测试集标注文件"
+                      name="test_file_path"
+                      rules={[
+                        {
+                          required: true,
+                          message: '请输入测试集标注文件',
+                        },
+                      ]}
+                    >
+                      <Input
+                        placeholder="请输入测试集标注文件"
+                        maxLength={64}
+                        showCount
+                        allowClear
+                      />
+                    </Form.Item>
+                  </Col>
+                </Row>
+              </>
+            );
+          }
+        }}
+      </Form.Item> */}
+    </>
+  );
+}
+
+export default VideoExecuteConfig;
diff --git a/react-ui/src/pages/AutoML/types.ts b/react-ui/src/pages/AutoML/types.ts
index 47b96974..8ca6f0fb 100644
--- a/react-ui/src/pages/AutoML/types.ts
+++ b/react-ui/src/pages/AutoML/types.ts
@@ -1,3 +1,11 @@
+/*
+ * @Author: error: error: git config user.name & please set dead value or install git && error: git config user.email & please set dead value or install git & please set dead value or install git
+ * @Date: 2025-04-17 10:22:05
+ * @LastEditors: error: error: git config user.name & please set dead value or install git && error: git config user.email & please set dead value or install git & please set dead value or install git
+ * @LastEditTime: 2025-04-27 11:43:19
+ * @FilePath: \ci4s\react-ui\src\pages\AutoML\types.ts
+ * @Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
+ */
 import { type ParameterInputObject } from '@/components/ResourceSelect';
 import { type NodeStatus } from '@/types';
 
@@ -43,6 +51,12 @@ export type FormData = {
   batch_size?: number;
   epochs?: number;
   lr?: number;
+  num_classes?: number;
+  is_validate?: boolean; 
+  train_data_prefix?: string; 
+  valid_data_prefix?: string; 
+  train_file_path?: string; 
+  valid_file_path?: string; 
 };
 
 export type AutoMLData = {