发布时不再筛去数据集

src/main/java/io/renren/modules/sys/controller/algs/algTrainController.java

@@ -414,12 +414,16 @@ public class algTrainController {
         List<ChannelSftp.LsEntry> models=new ArrayList<>();
         List<ChannelSftp.LsEntry> pythonFiles=new ArrayList<>();
         for(int i=0;i<vector.size();i++){
+            if (vector.get(i).getFilename().equals(".")||vector.get(i).getFilename().equals(".."))  continue;
             int lastPointLocation=vector.get(i).getFilename().lastIndexOf('.');
-            if(lastPointLocation==-1) continue;
+            if(lastPointLocation==-1) {
+                models.add(vector.get(i));
+                continue;
+            }
             //根据后缀名，判断属于哪类文件
             String suffix=vector.get(i).getFilename().substring(lastPointLocation);
             if(suffix.equals(".py"))    pythonFiles.add(vector.get(i));
-            else if(!suffix.equals(".")&&!suffix.equals(".csv")&&!suffix.equals(".out"))    models.add(vector.get(i));
+            else if(!suffix.equals(".")&&!suffix.equals(".out"))    models.add(vector.get(i));
         }
         return R.ok().put("algTrain",algTrain).put("models",models).put("pythonFiles",pythonFiles);
     }

src/main/java/io/renren/modules/sys/controller/algs/versionController.java

@@ -15,6 +15,7 @@ import io.renren.modules.sys.entity.algs.Algorithm_Vo;
 import io.renren.modules.sys.entity.algs.FileTest;
 import io.renren.modules.sys.entity.algs.Version;
 import io.renren.modules.sys.service.AlgsService;
+import io.renren.modules.sys.service.CategoryService;
 import io.renren.modules.sys.service.VersionService;
 import io.renren.modules.sys.service.impl.AlgsServiceImpl;
 import org.apache.shiro.authz.annotation.RequiresPermissions;
@@ -48,6 +49,9 @@ public class versionController {
     @Autowired
     AlgsService algsService;
 
+    @Autowired
+    CategoryService categoryService;
+
 
     /**
      * 所有列表
@@ -208,4 +212,19 @@ public class versionController {
         return R.ok();
     }
 
+    /**
+     * 根据algId和versionId查询对应信息
+     * @return
+     */
+    @GetMapping("/getStartTrainInfo")
+    public R getStartTrainInfo(@RequestParam Map<String,Object> params){
+        Long algId=Long.valueOf(String.valueOf(params.get("algId")));
+        Long versionId=Long.valueOf(String.valueOf(params.get("versionId")));
+        Algorithm alg=algsService.getById(algId);
+        String algorithmName=alg.getAlgorithmName();
+        String categoryName=categoryService.getById(alg.getCategoryId()).getCategoryName();
+        String versionName=versionService.getById(versionId).getVersionName();
+        return R.ok().put("algorithmName",algorithmName).put("categoryName",categoryName).put("versionName",versionName);
+    }
+
 }