AttributeImportanceFinderBuilder.java

package quickml.supervised.crossValidation.attributeImportance;

import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import quickml.data.AttributesMap;
import quickml.data.instances.InstanceWithAttributesMap;
import quickml.supervised.PredictiveModel;
import quickml.supervised.PredictiveModelBuilder;
import quickml.supervised.crossValidation.data.TrainingDataCycler;
import quickml.supervised.crossValidation.lossfunctions.classifierLossFunctions.ClassifierLossFunction;

import java.util.List;
import java.util.Set;

import static com.google.common.base.Preconditions.checkArgument;

public class AttributeImportanceFinderBuilder<T extends InstanceWithAttributesMap<?>> {


    private PredictiveModelBuilder<? extends PredictiveModel<AttributesMap, ?>, T> modelBuilder;
    private TrainingDataCycler<T> dataCycler;
    private double percentAttributesToRemovePerIteration = 0.2;
    private int numberOfIterations = 5;
    private Set<String> attributesToKeep = Sets.newHashSet();
    private List<ClassifierLossFunction> lossFunctions = Lists.newArrayList();
    private ClassifierLossFunction primaryLossFunction;


    public AttributeImportanceFinderBuilder<T> modelBuilder(PredictiveModelBuilder<? extends PredictiveModel<AttributesMap, ?>, T> modelBuilder) {
        this.modelBuilder = modelBuilder;
        return this;
    }

    public AttributeImportanceFinderBuilder<T> dataCycler(TrainingDataCycler<T> dataCycler) {
        this.dataCycler = dataCycler;
        return this;
    }

    public AttributeImportanceFinderBuilder<T> numOfIterations(int numberOfIterations) {
        this.numberOfIterations = numberOfIterations;
        return this;
    }

    public AttributeImportanceFinderBuilder<T> percentAttributesToRemovePerIteration(double attributesToRemovePerIteration) {
        this.percentAttributesToRemovePerIteration = attributesToRemovePerIteration;
        return this;
    }

    public AttributeImportanceFinderBuilder<T> primaryLossFunction(ClassifierLossFunction primaryLossFunction) {
        lossFunctions.add(primaryLossFunction);
        this.primaryLossFunction = primaryLossFunction;
        return this;
    }

    public AttributeImportanceFinderBuilder<T> lossFunction(ClassifierLossFunction lossFunction) {
        this.lossFunctions.add(lossFunction);
        return this;
    }

    public AttributeImportanceFinderBuilder<T> attributesToKeep(Set<String> attributesToKeep) {
        this.attributesToKeep = attributesToKeep;
        return this;
    }

    public AttributeImportanceFinder<T> build() {
        checkArgument(primaryLossFunction != null, "A primary loss function must be set");
        checkArgument(modelBuilder != null, "Must supply a model builder");
        checkArgument(dataCycler != null, "Must supply a data cycler");

        return new AttributeImportanceFinder<>(modelBuilder, dataCycler, percentAttributesToRemovePerIteration,
                numberOfIterations, attributesToKeep, lossFunctions, primaryLossFunction);
    }


}