我有一份未评级的商品清单,我希望将它们分成5个折叠。目前,我已经完成了,因为我只是尝试均匀地分成块,最后一个被称为"扩展"大块,这会给我带来麻烦。
当我有nonRatedItems = {1,2,3,4,5,6,7,8}
时,那些块是:
1
2
3
4
5 6 7 8
我想避免这种情况,以便输出为:
1 6
2 7
3 8
4
5
当前来源:
Collections.shuffle(nonRatedItems);
nonRatedItems = nonRatedItems.subList(0, (int) Math.ceil(nonRatedItems.size() * 0.2));
int tfoldSize = nonRatedItems.size() / 5;
List<List<Integer>> negativeTestFolds = new ArrayList<List<Integer>>();
if (tfoldSize < 1) {
nonRatedItems.stream().map(Lists::newArrayList).collect(Collectors.toCollection(() -> negativeTestFolds));
} else {
for (int i = 0; i < 5; i++) {
int endIndex = i < 4 ? (i + 1) * tfoldSize : nonRatedItems.size();
negativeTestFolds.add(nonRatedItems.subList(i * tfoldSize, endIndex));
}
}
感谢任何建议/帮助
答案 0 :(得分:2)
总是5折吗? 为什么不循环列表?
之类的东西List<Integer>[] folds = new List<Integer>[5];
for(int i = 0; i < 5; i++)
{
folds[i] = new LinkedList<Integer>();
}
for (int i = 0; i < nonRatedItems.size(); i++)
{
folds[i % 5].add(nonRatedItems.get(i));
}