Browse Source

feat: 添加底层模型获取功能并更新相关逻辑

pull/41/head
Jiale 5 months ago
parent
commit
99e38cf13d
  1. 4
      src/api/alert/model/models.ts
  2. 33
      src/views/model/train/index.vue

4
src/api/alert/model/models.ts

@ -10,6 +10,7 @@ enum Api {
OPTIMISTIC = '/alert/optimistic', OPTIMISTIC = '/alert/optimistic',
TRAIN_MODEL = '/alert/model/train', TRAIN_MODEL = '/alert/model/train',
TEST_MODEL = '/alert/model/test', TEST_MODEL = '/alert/model/test',
BOTTOM_MODEL = '/alert/model/bottom/',
} }
export function modelCardListApi(params?: ModelQueryParams) { export function modelCardListApi(params?: ModelQueryParams) {
return defHttp.get<ModelCardItem[]>({ url: Api.MODEL_CARD_LIST, params }) return defHttp.get<ModelCardItem[]>({ url: Api.MODEL_CARD_LIST, params })
@ -35,4 +36,7 @@ export function calculateBackApi(id: any, params: any) {
export const getOptimisticApi = (params: any) => defHttp.get<any>({ url: Api.OPTIMISTIC, params }) export const getOptimisticApi = (params: any) => defHttp.get<any>({ url: Api.OPTIMISTIC, params })
export const trainModelApi = (params: any) => defHttp.post<any>({ url: Api.TRAIN_MODEL, data: params }) export const trainModelApi = (params: any) => defHttp.post<any>({ url: Api.TRAIN_MODEL, data: params })
export const testModelApi = (params: any) => defHttp.post<any>({ url: Api.TEST_MODEL, data: params }) export const testModelApi = (params: any) => defHttp.post<any>({ url: Api.TEST_MODEL, data: params })
export const bottomModelApi = (id: any) => defHttp.post<any>({ url: Api.BOTTOM_MODEL + id })

33
src/views/model/train/index.vue

@ -29,6 +29,7 @@ import { pointTableSchema, sampleInfoTableSchema } from './data'
import { BasicTable, useTable } from '@/components/Table' import { BasicTable, useTable } from '@/components/Table'
import { PageWrapper } from '@/components/Page' import { PageWrapper } from '@/components/Page'
import { import {
bottomModelApi,
modelInfoApi, modelInfoApi,
testModelApi, testModelApi,
trainModelApi, trainModelApi,
@ -140,7 +141,7 @@ export default defineComponent({
async function getTestData() { async function getTestData() {
const params = { const params = {
Model_id: 118, Model_id: 530,
version: model.value?.Cur_Version ? model.value?.Cur_Version : 'v-test', version: model.value?.Cur_Version ? model.value?.Cur_Version : 'v-test',
Test_Data: { Test_Data: {
time: historyTime.value time: historyTime.value
@ -479,16 +480,22 @@ export default defineComponent({
} }
function openPointModal(index, record) { function openPointModal(index, record) {
// index
const pageIndex = index
// index
const globalIndex = model.value.pointInfo.findIndex(
item => item.pointId === record.pointId,
)
openEditPointModal.value = true openEditPointModal.value = true
pointEditRecord = record pointEditRecord = record
editForm.value = { editForm.value = {
index, index: globalIndex,
Upper: record?.Upper ?? '', Upper: record?.Upper ?? '',
Lower: record?.Lower ?? '', Lower: record?.Lower ?? '',
lowerBound: record?.lowerBound ?? '', lowerBound: record?.lowerBound ?? '',
upperBound: record?.upperBound ?? '', upperBound: record?.upperBound ?? '',
dead: record?.dead ?? false, dead: !!record?.dead,
limit: record?.limit ?? true, limit: !!record?.limit,
} }
} }
@ -595,6 +602,22 @@ export default defineComponent({
openEditModelModal.value = false openEditModelModal.value = false
} }
async function bottomModel() {
if (!model.value.para) {
createMessage.error('模型未训练,无法下装')
return
}
try {
const response = await bottomModelApi(model.value.id)
model.value = response
createMessage.success('模型下装成功')
}
catch (error) {
console.error('底层模型获取失败:', error)
createMessage.error('底层模型获取失败')
}
}
return { return {
pointTable, pointTable,
model, model,
@ -629,6 +652,7 @@ export default defineComponent({
openEditModel, openEditModel,
handleEditModelOk, handleEditModelOk,
handleEditModelCancel, handleEditModelCancel,
bottomModel,
} }
}, },
}) })
@ -767,6 +791,7 @@ export default defineComponent({
<a-button <a-button
danger danger
style="margin-left: 10px" style="margin-left: 10px"
@click="bottomModel"
> >
下装 下装
</a-button> </a-button>

Loading…
Cancel
Save