com.kogalur.randomforest

Class RandomForest

java.lang.Object

com.kogalur.randomforest.RandomForest

public class RandomForest
extends Object

Class that provides the static methods to train a random forest, to restore a previouly created random forest, and to predict using new test data with a previously created random forest.

Author:

Udaya Kogalur

Method Summary

Modifier and Type	Method and Description
static RandomForestModel	predict(ModelArg modelArg, org.apache.spark.sql.Dataset dataset)
static RandomForestModel	predict(RandomForestModel model) Restores a random forest given the user-defined model arugments.
static RandomForestModel	train(ModelArg modelArg) Trains a random forest given the user-defined model arugments.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

train

public static RandomForestModel train(ModelArg modelArg)

Trains a random forest given the user-defined model arugments.

Parameters:

modelArg - User-defined model arguments.

predict

public static RandomForestModel predict(RandomForestModel model)

Restores a random forest given the user-defined model arugments.

Parameters:

model - User-defined model arguments.

predict

public static RandomForestModel predict(ModelArg modelArg,
                                        org.apache.spark.sql.Dataset dataset)