Browse Source

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

pull/41/head
Jiale 3 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',
TRAIN_MODEL = '/alert/model/train',
TEST_MODEL = '/alert/model/test',
BOTTOM_MODEL = '/alert/model/bottom/',
}
export function modelCardListApi(params?: ModelQueryParams) {
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 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 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 { PageWrapper } from '@/components/Page'
import {
bottomModelApi,
modelInfoApi,
testModelApi,
trainModelApi,
@ -140,7 +141,7 @@ export default defineComponent({
async function getTestData() {
const params = {
Model_id: 118,
Model_id: 530,
version: model.value?.Cur_Version ? model.value?.Cur_Version : 'v-test',
Test_Data: {
time: historyTime.value
@ -479,16 +480,22 @@ export default defineComponent({
}
function openPointModal(index, record) {
// index
const pageIndex = index
// index
const globalIndex = model.value.pointInfo.findIndex(
item => item.pointId === record.pointId,
)
openEditPointModal.value = true
pointEditRecord = record
editForm.value = {
index,
index: globalIndex,
Upper: record?.Upper ?? '',
Lower: record?.Lower ?? '',
lowerBound: record?.lowerBound ?? '',
upperBound: record?.upperBound ?? '',
dead: record?.dead ?? false,
limit: record?.limit ?? true,
dead: !!record?.dead,
limit: !!record?.limit,
}
}
@ -595,6 +602,22 @@ export default defineComponent({
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 {
pointTable,
model,
@ -629,6 +652,7 @@ export default defineComponent({
openEditModel,
handleEditModelOk,
handleEditModelCancel,
bottomModel,
}
},
})
@ -767,6 +791,7 @@ export default defineComponent({
<a-button
danger
style="margin-left: 10px"
@click="bottomModel"
>
下装
</a-button>

Loading…
Cancel
Save