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