• Home
• com.flipkart.fdp.ml
• models-info
• 0.1.2
• source code
• RandomForestModelInfo.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.flipkart.fdp.ml.modelinfo.RandomForestModelInfo Maven / Gradle / Ivy

package com.flipkart.fdp.ml.modelinfo;

import com.flipkart.fdp.ml.transformer.RandomForestTransformer;
import com.flipkart.fdp.ml.transformer.Transformer;
import lombok.Data;

import java.util.ArrayList;

/**
 * Represents information for a Random Forest model
 */

@Data
public class RandomForestModelInfo extends AbstractModelInfo {
    private String algorithm;
    private ArrayList trees = new ArrayList();

    /**
     * @return an corresponding {@link RandomForestTransformer} for this model info
     */
    @Override
    public Transformer getTransformer() {
        return new RandomForestTransformer(this);
    }
}