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