295 lines
7.8 KiB
Vue
295 lines
7.8 KiB
Vue
<template>
|
|
<div class="prediction-view">
|
|
<el-card>
|
|
<template #header>
|
|
<div class="card-header">
|
|
<span>按店铺预测</span>
|
|
<el-tooltip content="使用针对特定店铺训练的模型进行销售预测">
|
|
<el-icon><QuestionFilled /></el-icon>
|
|
</el-tooltip>
|
|
</div>
|
|
</template>
|
|
|
|
<div class="model-selection-section">
|
|
<h4>🎯 选择预测模型</h4>
|
|
<el-form :model="form" label-width="120px">
|
|
<el-row :gutter="20">
|
|
<el-col :span="8">
|
|
<el-form-item label="目标店铺">
|
|
<StoreSelector
|
|
v-model="form.store_id"
|
|
@change="handleStoreChange"
|
|
:show-all-option="false"
|
|
/>
|
|
</el-form-item>
|
|
</el-col>
|
|
<el-col :span="8">
|
|
<el-form-item label="算法类型">
|
|
<el-select
|
|
v-model="form.model_type"
|
|
placeholder="选择算法"
|
|
@change="handleModelTypeChange"
|
|
style="width: 100%"
|
|
:disabled="!form.store_id"
|
|
>
|
|
<el-option
|
|
v-for="item in modelTypes"
|
|
:key="item.id"
|
|
:label="item.name"
|
|
:value="item.id"
|
|
/>
|
|
</el-select>
|
|
</el-form-item>
|
|
</el-col>
|
|
</el-row>
|
|
|
|
<el-row :gutter="20" v-if="form.model_type">
|
|
<el-col :span="6">
|
|
<el-form-item label="模型版本">
|
|
<el-select
|
|
v-model="form.version"
|
|
placeholder="选择版本"
|
|
style="width: 100%"
|
|
:disabled="!availableVersions.length"
|
|
:loading="versionsLoading"
|
|
>
|
|
<el-option
|
|
v-for="version in availableVersions"
|
|
:key="version"
|
|
:label="version"
|
|
:value="version"
|
|
/>
|
|
</el-select>
|
|
</el-form-item>
|
|
</el-col>
|
|
<el-col :span="6">
|
|
<el-form-item label="预测天数">
|
|
<el-input-number
|
|
v-model="form.future_days"
|
|
:min="1"
|
|
:max="365"
|
|
style="width: 100%"
|
|
/>
|
|
</el-form-item>
|
|
</el-col>
|
|
<el-col :span="6">
|
|
<el-form-item label="起始日期">
|
|
<el-date-picker
|
|
v-model="form.start_date"
|
|
type="date"
|
|
placeholder="选择日期"
|
|
format="YYYY-MM-DD"
|
|
value-format="YYYY-MM-DD"
|
|
style="width: 100%"
|
|
:clearable="false"
|
|
/>
|
|
</el-form-item>
|
|
</el-col>
|
|
<el-col :span="6">
|
|
<el-form-item label="预测分析">
|
|
<el-switch
|
|
v-model="form.analyze_result"
|
|
active-text="开启"
|
|
inactive-text="关闭"
|
|
/>
|
|
</el-form-item>
|
|
</el-col>
|
|
</el-row>
|
|
</el-form>
|
|
</div>
|
|
|
|
<div class="prediction-actions">
|
|
<el-button
|
|
type="primary"
|
|
size="large"
|
|
@click="startPrediction"
|
|
:loading="predicting"
|
|
:disabled="!canPredict"
|
|
>
|
|
<el-icon><TrendCharts /></el-icon>
|
|
开始预测
|
|
</el-button>
|
|
</div>
|
|
</el-card>
|
|
|
|
<el-card v-if="predictionResult" style="margin-top: 20px">
|
|
<template #header>
|
|
<div class="card-header">
|
|
<span>📈 预测结果</span>
|
|
</div>
|
|
</template>
|
|
<div class="prediction-chart">
|
|
<canvas ref="chartCanvas" width="800" height="400"></canvas>
|
|
</div>
|
|
</el-card>
|
|
</div>
|
|
</template>
|
|
|
|
<script setup>
|
|
import { ref, reactive, onMounted, computed, watch, nextTick } from 'vue'
|
|
import axios from 'axios'
|
|
import { ElMessage } from 'element-plus'
|
|
import { QuestionFilled, TrendCharts } from '@element-plus/icons-vue'
|
|
import Chart from 'chart.js/auto'
|
|
import StoreSelector from '../../components/StoreSelector.vue'
|
|
|
|
const modelTypes = ref([])
|
|
const availableVersions = ref([])
|
|
const versionsLoading = ref(false)
|
|
const predicting = ref(false)
|
|
const predictionResult = ref(null)
|
|
const chartCanvas = ref(null)
|
|
let chart = null
|
|
|
|
const form = reactive({
|
|
training_mode: 'store',
|
|
store_id: '',
|
|
model_type: '',
|
|
version: '',
|
|
future_days: 7,
|
|
start_date: '',
|
|
analyze_result: true
|
|
})
|
|
|
|
const canPredict = computed(() => {
|
|
return form.store_id && form.model_type && form.version
|
|
})
|
|
|
|
const fetchModelTypes = async () => {
|
|
try {
|
|
const response = await axios.get('/api/model_types')
|
|
if (response.data.status === 'success') {
|
|
modelTypes.value = response.data.data
|
|
}
|
|
} catch (error) {
|
|
ElMessage.error('获取模型类型失败')
|
|
}
|
|
}
|
|
|
|
const fetchAvailableVersions = async () => {
|
|
if (!form.store_id || !form.model_type) {
|
|
availableVersions.value = []
|
|
return
|
|
}
|
|
try {
|
|
versionsLoading.value = true
|
|
const url = `/api/models/store/${form.store_id}/${form.model_type}/versions`
|
|
const response = await axios.get(url)
|
|
if (response.data.status === 'success') {
|
|
availableVersions.value = response.data.data.versions || []
|
|
if (response.data.data.latest_version) {
|
|
form.version = response.data.data.latest_version
|
|
}
|
|
}
|
|
} catch (error) {
|
|
availableVersions.value = []
|
|
} finally {
|
|
versionsLoading.value = false
|
|
}
|
|
}
|
|
|
|
const handleStoreChange = () => {
|
|
form.model_type = ''
|
|
form.version = ''
|
|
availableVersions.value = []
|
|
}
|
|
|
|
const handleModelTypeChange = () => {
|
|
form.version = ''
|
|
fetchAvailableVersions()
|
|
}
|
|
|
|
const startPrediction = async () => {
|
|
try {
|
|
predicting.value = true
|
|
const payload = {
|
|
model_type: form.model_type,
|
|
version: form.version,
|
|
future_days: form.future_days,
|
|
start_date: form.start_date,
|
|
analyze_result: form.analyze_result,
|
|
store_id: form.store_id
|
|
}
|
|
const response = await axios.post('/api/predict', payload)
|
|
if (response.data.status === 'success') {
|
|
predictionResult.value = response.data.data
|
|
ElMessage.success('预测完成！')
|
|
await nextTick()
|
|
renderChart()
|
|
} else {
|
|
ElMessage.error(response.data.message || '预测失败')
|
|
}
|
|
} catch (error) {
|
|
ElMessage.error('预测请求失败')
|
|
} finally {
|
|
predicting.value = false
|
|
}
|
|
}
|
|
|
|
const renderChart = () => {
|
|
if (!chartCanvas.value || !predictionResult.value) return
|
|
if (chart) {
|
|
chart.destroy()
|
|
}
|
|
const predictions = predictionResult.value.predictions
|
|
const labels = predictions.map(p => p.date)
|
|
const data = predictions.map(p => p.sales)
|
|
chart = new Chart(chartCanvas.value, {
|
|
type: 'line',
|
|
data: {
|
|
labels,
|
|
datasets: [{
|
|
label: '预测销量',
|
|
data,
|
|
borderColor: '#409EFF',
|
|
backgroundColor: 'rgba(64, 158, 255, 0.1)',
|
|
tension: 0.4,
|
|
fill: true
|
|
}]
|
|
},
|
|
options: {
|
|
responsive: true,
|
|
plugins: {
|
|
title: {
|
|
display: true,
|
|
text: '销量预测趋势图'
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
onMounted(() => {
|
|
fetchModelTypes()
|
|
const today = new Date()
|
|
form.start_date = today.toISOString().split('T')[0]
|
|
})
|
|
|
|
watch([() => form.store_id, () => form.model_type], () => {
|
|
fetchAvailableVersions()
|
|
})
|
|
</script>
|
|
|
|
<style scoped>
|
|
.prediction-view {
|
|
padding: 20px;
|
|
}
|
|
.card-header {
|
|
display: flex;
|
|
justify-content: space-between;
|
|
align-items: center;
|
|
}
|
|
.model-selection-section h4 {
|
|
margin-bottom: 16px;
|
|
}
|
|
.prediction-actions {
|
|
display: flex;
|
|
justify-content: center;
|
|
margin-top: 20px;
|
|
padding-top: 20px;
|
|
border-top: 1px solid #ebeef5;
|
|
}
|
|
.prediction-chart {
|
|
margin-top: 20px;
|
|
}
|
|
</style> |