ae.teletronics.nlp.entityextraction.types.sender

Class DecisionTreeTrainer

java.lang.Object

ae.teletronics.nlp.entityextraction.types.sender.DecisionTreeTrainer

All Implemented Interfaces:

Trainer

public class DecisionTreeTrainer
extends Object
implements Trainer

Created by hhravn on 06/06/16.

Constructor Summary

Constructors

Constructor and Description
DecisionTreeTrainer()

Method Summary

Modifier and Type	Method and Description
DecisionTreeModel	load(org.apache.spark.SparkContext sc, String fileName)
String	name()
DecisionTreeModel	train(org.apache.spark.rdd.RDD<org.apache.spark.mllib.regression.LabeledPoint> data)

Methods inherited from class java.lang.Object

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

Constructor Detail

DecisionTreeTrainer

public DecisionTreeTrainer()

Method Detail

train

public DecisionTreeModel train(org.apache.spark.rdd.RDD<org.apache.spark.mllib.regression.LabeledPoint> data)

Specified by:

train in interface Trainer

name

public String name()

Specified by:

name in interface Trainer

load

public DecisionTreeModel load(org.apache.spark.SparkContext sc,
                              String fileName)

Specified by:

load in interface Trainer