优化yaml导出测试

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java

@@ -557,7 +557,15 @@ public class ExperimentServiceImpl implements ExperimentService {
 
             //处理指标
             HashMap<String, Object> metricMap = JSON.parseObject(metricRecord, HashMap.class);
-            for (Map.Entry<String, Object> metric : metricMap.entrySet()) {
+            HashMap<String, Object> trainMetric = (HashMap<String, Object>)metricMap.get("train");
+            HashMap<String, Object> evalMetric = (HashMap<String, Object>)metricMap.get("evaluate");
+            for (Map.Entry<String, Object> metric : trainMetric.entrySet()) {
+                if (YamlUtils.isNumeric(String.valueOf(metric.getValue()))) {
+                    BigDecimal bigDecimal = new BigDecimal(String.valueOf(metric.getValue()));
+                    metric.setValue(bigDecimal);
+                }
+            }
+            for (Map.Entry<String, Object> metric : evalMetric.entrySet()) {
                 if (YamlUtils.isNumeric(String.valueOf(metric.getValue()))) {
                     BigDecimal bigDecimal = new BigDecimal(String.valueOf(metric.getValue()));
                     metric.setValue(bigDecimal);