Просмотр исходного кода

show model in modelDetail & train

icenaked 3 лет назад
Родитель
Сommit
b0c4e53f6a
2 измененных файлов с 33 добавлено и 7 удалено
  1. 3 3
      src/api/model.js
  2. 30 4
      src/views/unused/ModelDetail.vue

+ 3 - 3
src/api/model.js

@@ -48,8 +48,8 @@ export const deleteModel=async modelId=>{
      * @returns
      */
     export const getAllModel = async userId => {
-        const res = await axios.get(`${MODEL_MODULE}/get/all`, {params: {userId: userId}})
-        return res.data//response
+        const res = await axios.get(`${MODEL_MODULE}/get/all`, {params:  userId})
+        return res//response
     }
     /**
      * 获取深度学习模型 GET /model/get/all/dl
@@ -95,7 +95,7 @@ export const deleteModel=async modelId=>{
      * @returns
      */
     export const train = async modelId=>{
-        const res=await axios.get(`${MODEL_MODULE}/train`,{params:{modelId:modelId}});
+        const res=await axios.get(`${MODEL_MODULE}/train`,{params:modelId});
         return res;
     }
     /**

+ 30 - 4
src/views/unused/ModelDetail.vue

@@ -31,7 +31,9 @@
                 @row-dblclick="getConfigDetail"
             >
                 <el-table-column width="50" type="selection"></el-table-column>
-                <el-table-column prop="confName" label="模型Id"></el-table-column>
+                <el-table-column prop="confName" label="模型名">
+                  <template #default="scope">{{ scope.row.modelName }}</template>
+                </el-table-column>
                 <el-table-column label="模型信息">
                     <template #default="scope">{{ scope.row.modelTypeName }}</template>
                 </el-table-column>
@@ -47,7 +49,7 @@
                 <el-table-column label="模型训练" width="180" align="center">
                     <template #default="scope">
                         <el-button type="text" icon="el-icon-video-play" class="green"
-                            @click="handleDelete(scope.$index, scope.row)">模型训练</el-button>
+                            @click="handleTrain(scope.row.modelId)">模型训练</el-button>
                     </template>
                 </el-table-column>
                 <!-- 如果已经训练了,那么就告诉她您已经训练过该模型,但还是感觉应该放在模型配置里面,配置就可以让他进行训练! -->
@@ -121,6 +123,7 @@ import { ref, reactive } from "vue";
 import { ElMessage, ElMessageBox } from "element-plus";
 import { fetchData } from "../../api";
 import {addConfig, deleteModelByModelId, getAllConfig} from "../../api/config";
+import {getAllModel, train} from "../../api/model";
 
 export default {
     name: "config",
@@ -165,8 +168,10 @@ export default {
 
     const getpage=()=> {
       // console.log("getpage");
-      getAllConfig({
-        uid: 1
+      //getAllConfig({
+        //uid: 1
+      getAllModel({
+        userId: 1
       }).then(res => {
         console.log("res",res);
         tableData.value = res.data;
@@ -188,6 +193,26 @@ export default {
             getData();
         };
 
+        const handleTrain =(modelId) => {
+          ElMessageBox.confirm("确定要训练吗?", "提示", {
+              type: "warning",
+          })
+              .then(() => {
+                  // ElMessage.success("删除成功");
+                //tableData.value.splice(index, 1);
+                //let idx=tableData._rawValue[index].id;
+                let idx=modelId;
+                console.log("modelId: "+idx);
+                train({
+                  modelId: idx
+                }).then(res=>{
+                  console.log("train_res: ",res);
+                });
+
+              })
+              .catch(() => {});
+        };
+
         // 删除操作(单个)
         const handleDelete = (index) => {
           ElMessage.success("正在训练中...")
@@ -295,6 +320,7 @@ export default {
             form,
             handleSearch,
             handlePageChange,
+          handleTrain,
             handleDelete,
           handleSelectionChange,
             handleBatchDelete,