RFRequestImpl.java

/*
 * Copyright 2013 State University of New York at Oswego
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); 
 * you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
 * See the License for the specific language governing permissions and 
 * limitations under the License.
 */
package edu.oswego.csc480_hci521_2013.shared.h2o;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

// TODO: Merge with shared.h2o.urlbuilders.RFBuilder    
public class RFRequestImpl implements RFRequest {

    private String dataKey;
    private String responseVariable;
    private int numberTrees;
    private boolean gini;
    private boolean oob;
    private boolean stratify;
    private int numberFeatures;
    private List<String> features = new ArrayList<String>();
    private String classVariable;

    public RFRequestImpl(String key, List<String> features) {
        this.dataKey = key;
        this.numberTrees = 1000;
        this.features = features;

        if (!this.features.isEmpty()) {
            this.classVariable = this.features.get(0);
        }
    }

    public String getDataKey() {
        return this.dataKey;
    }

    public void setDataKey(String dataKey) {
        this.dataKey = dataKey;
    }

    public String getResponseVariable() {
        return this.responseVariable;
    }

    public void setResponseVariable(String responseVariable) {
        this.responseVariable = responseVariable;
    }

    public int getNumberTrees() {
        return this.numberTrees;
    }

    public void setNumberTrees(int nTrees) {
        this.numberTrees = nTrees;
    }

    public boolean getGini() {
        return this.gini;
    }

    public void setGini(boolean gini) {
        this.gini = gini;
    }

    public boolean getOutOfBagErrors() {
        return this.oob;
    }

    public void setOutOfBagerrors(boolean oob) {
        this.oob = oob;
    }
    
    public boolean getStratify() {
        return this.stratify;
    }

    public void setStratify(boolean stratify) {
        this.stratify = stratify;
    }
    
    public int getNumberFeatures() {
        return this.numberFeatures;
    }

    public void setNumberFeatures(int numberFeatures) {
        this.numberFeatures = numberFeatures;
    }

    public List<String> getFeatures() {
        return this.features;
    }

    public void setFeatures(List<String> features) {
        this.features = features;
    }

    public String getClassVariable() {
        return this.classVariable;
    }

    public void setClassVariable(String classVariable) {
        this.classVariable = classVariable;
    }

    public boolean IsValid() {
        if (this.numberTrees <= 0) {
            return false;
        }

        if ((this.numberFeatures > 0) && (this.numberFeatures <= this.features.size())) {
            return false;
        }

        if (!this.features.contains(this.classVariable)) {
            return false;
        }

        return true;
    }

    @Override
    public HashMap<String, Double> getClassWeights() {
        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
    }

    @Override
    public void setClassWeights(HashMap<String, Double> values) {
        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
    }
}