浏览代码

发布算法不再筛去数据集

wangyi 2 年之前
父节点
当前提交
cafddf2676
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      src/main/java/io/renren/modules/sys/controller/algs/algTrainController.java

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

@@ -422,8 +422,8 @@ public class algTrainController {
             }
             //根据后缀名,判断属于哪类文件
             String suffix=vector.get(i).getFilename().substring(lastPointLocation);
-            if(suffix.equals(".py"))    pythonFiles.add(vector.get(i));
-            else if(!suffix.equals(".")&&!suffix.equals(".out"))    models.add(vector.get(i));
+            if(suffix.equals(".py")||suffix.equals(".csv"))    pythonFiles.add(vector.get(i));
+            else if(!suffix.equals(".")&&!suffix.equals(".csv")&&!suffix.equals(".out"))    models.add(vector.get(i));
         }
         return R.ok().put("algTrain",algTrain).put("models",models).put("pythonFiles",pythonFiles);
     }