修改一些小BUG

application-webadmin/src/main/java/com/yice/webadmin/app/controller/ModelEstimateController.java

@@ -95,11 +95,12 @@ public class ModelEstimateController {
         ModelEstimate modelEstimate = this.modelEstimateService.getById(taskId);
         ModelVersion modelVersion = this.modelVersionService.getById(modelEstimate.getModelVersionId());
         DatasetVersion datasetVersion = this.datasetVersionService.getById(modelEstimate.getDatasetVersionId());
-        ModelManage modelManage = this.modelManageService.getById(modelVersion.getModelId());
         JSONObject jsonObject = (JSONObject) JSON.parse(modelEstimate.getConfiguration());
+        JSONArray datasetVersionNames = new JSONArray();
+        datasetVersionNames.add(datasetVersion.getVersionName());
         JSONArray array = new JSONArray();
         array.add("zh");
-        array.add(modelManage.getModelName());
+        array.add(modelVersion.getVersionName());
         array.add(modelVersion.getModelUrl());
         array.add("");
         array.add(new JSONArray());
@@ -110,7 +111,7 @@ public class ModelEstimateController {
         array.add(false);
         array.add("none");
         array.add(pythonConfig.getDatasetFileMenu());
-        array.add(datasetVersion.getVersionName());
+        array.add(datasetVersionNames);
         array.add(jsonObject.get("truncationLength"));
         array.add(jsonObject.get("maximumSampleSize"));
         array.add(jsonObject.get("batchSize"));

application-webadmin/src/main/java/com/yice/webadmin/app/controller/TuningRunController.java

@@ -130,14 +130,12 @@ public class TuningRunController {
         TuningRun tuningRun = this.tuningRunService.getById(runId);
         ModelVersion modelVersion = this.modelVersionService.getById(tuningRun.getModelVersionId());
         DatasetVersion datasetVersion = this.datasetVersionService.getById(tuningRun.getDatasetVersionId());
-        DatasetManage datasetManage = this.datasetManageService.getById(datasetVersion.getDatasetId());
-        ModelManage modelManage = this.modelManageService.getById(modelVersion.getModelId());
         JSONObject jsonObject = (JSONObject) JSON.parse(tuningRun.getConfiguration());
         JSONArray datasetVersionNames = new JSONArray();
-        datasetVersionNames.add(datasetManage.getDatasetName() + "_V" + datasetVersion.getDatasetVersion());
+        datasetVersionNames.add(datasetVersion.getVersionName());
         JSONArray array = new JSONArray();
         array.add("zh");
-        array.add(modelManage.getModelName());
+        array.add(modelVersion.getVersionName());
         array.add(modelVersion.getModelUrl());
         array.add(tuningRun.getTrainMethod());
         array.add(new JSONArray());