com.expleague.ml.methods.trees.MutedFeatureObliviousTreeEnsemble Maven / Gradle / Ivy

Go to download
package com.expleague.ml.methods.trees;

import com.expleague.commons.math.vectors.impl.vectors.ArrayVec;
import org.jetbrains.annotations.NotNull;

import java.util.Arrays;


import com.expleague.commons.math.vectors.Vec;
import com.expleague.ml.func.Ensemble;
import com.expleague.ml.models.ObliviousTree;

/**
 * Created by vkokarev on 09.12.14.
 */
public class MutedFeatureObliviousTreeEnsemble extends Ensemble {
  private MutedFeatureObliviousTreeEnsemble(final ObliviousTree[] models, final Vec weights) {
    super(models, weights);
  }

  public static MutedFeatureObliviousTreeEnsemble from(@NotNull final Ensemble origin, @NotNull final int ... mutedFeaturesIndexes) throws ClassCastException {
    final ObliviousTree[] newModels = new ObliviousTree[origin.models.length];
    final Vec weights = new ArrayVec(origin.weights.length());
    int realIdx = 0;
    for (int i = 0; i < origin.models.length; i++) {
      final ObliviousTree model = ObliviousTree.removeFeatures(origin.models[i], mutedFeaturesIndexes);
      if (model != null) {
        newModels[realIdx] = model;
        weights.set(realIdx++, origin.weights.get(i));
      }
    }
    return new MutedFeatureObliviousTreeEnsemble(Arrays.copyOf(newModels, realIdx), new ArrayVec(weights.toArray(), 0, realIdx));
  }
}