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);