千家信息网

predictionio 随机森林

发表于:2025-01-31 作者:千家信息网编辑
千家信息网最后更新 2025年01月31日,其实predictionio 也是支持随机森林算法的,可惜官方的例子没有出现,我简单写了一点,您也可以参考这个地址:https://github.com/runapal5/PredictionIO-D
千家信息网最后更新 2025年01月31日predictionio 随机森林

其实predictionio 也是支持随机森林算法的,可惜官方的例子没有出现,我简单写了一点,您也可以参考这个地址:
https://github.com/runapal5/PredictionIO-DefectPrediction/blob/master/engine.json

public class RandomForestAlgorithm extends P2LJavaAlgorithm {    /**     *      */    private static final long serialVersionUID = 1L;    private final RandomForestAlgorithmParams ap;    public RandomForestAlgorithm(RandomForestAlgorithmParams ap) {        this.ap = ap;    }    @Override    public RandomForestModel train(SparkContext sc, PreparedData preparedData) {        String datapath = "/root/whk/company.txt";        JavaRDD treedata = MLUtils.loadLibSVMFile(sc, datapath).toJavaRDD();        Map categoricalFeaturesInfo = new HashMap();        return RandomForest.trainClassifier(treedata, ap.getNumClasses(), categoricalFeaturesInfo, ap.getNumTrees(),                                     ap.getFeatureSubsetStrategy(),ap.getImpurity(),ap.getMaxDepth(),ap.getMaxBins(),12345);    }    @Override    public PredictedResult predict(RandomForestModel model, final Query query) {        double[] b={12,0,0};        model.predict(Vectors.dense(b));                                          return null;    }}
0