From 61eb75a3f7a4690d036cbfc15041a443828dcf12 Mon Sep 17 00:00:00 2001 From: chenzhihang <709011834@qq.com> Date: Fri, 27 Sep 2024 15:59:45 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96yaml=E5=AF=BC=E5=87=BA?= =?UTF-8?q?=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../platform/service/impl/ExperimentServiceImpl.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java index 16af4e8c..1e804faf 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java @@ -557,7 +557,15 @@ public class ExperimentServiceImpl implements ExperimentService { //处理指标 HashMap metricMap = JSON.parseObject(metricRecord, HashMap.class); - for (Map.Entry metric : metricMap.entrySet()) { + HashMap trainMetric = (HashMap)metricMap.get("train"); + HashMap evalMetric = (HashMap)metricMap.get("evaluate"); + for (Map.Entry metric : trainMetric.entrySet()) { + if (YamlUtils.isNumeric(String.valueOf(metric.getValue()))) { + BigDecimal bigDecimal = new BigDecimal(String.valueOf(metric.getValue())); + metric.setValue(bigDecimal); + } + } + for (Map.Entry metric : evalMetric.entrySet()) { if (YamlUtils.isNumeric(String.valueOf(metric.getValue()))) { BigDecimal bigDecimal = new BigDecimal(String.valueOf(metric.getValue())); metric.setValue(bigDecimal);