com.expleague.ml.data.tools.SubPool Maven / Gradle / Ivy
package com.expleague.ml.data.tools;
import com.expleague.commons.seq.Seq;
import com.expleague.commons.util.ArrayTools;
import com.expleague.commons.util.Pair;
import com.expleague.ml.meta.DSItem;
import com.expleague.ml.meta.PoolFeatureMeta;
/**
* User: solar
* Date: 11.07.14
* Time: 22:57
*/
public class SubPool extends Pool {
public SubPool(final Pool original, final int[] indices) {
super(original.meta,
ArrayTools.cut(original.items, indices),
cutFeatures(original.features, indices),
cutFeatures(original.targets.toArray(new Pair[original.targets.size()]), indices));
}
private static Pair>[] cutFeatures(final Pair>[] original, final int[] indices) {
@SuppressWarnings("unchecked")
final Pair>[] result = new Pair[original.length];
for (int i = 0; i < original.length; i++) {
result[i] = Pair.>create(original[i].first, ArrayTools.cut(original[i].getSecond(), indices));
}
return result;
}
}