co.cask.hydrator.plugin.spark

Class DecisionTreeTrainer.DecisionTreeTrainerConfig

java.lang.Object

co.cask.cdap.api.Config

co.cask.cdap.api.plugin.PluginConfig

co.cask.hydrator.plugin.spark.SparkMLTrainer.MLTrainerConfig

co.cask.hydrator.plugin.spark.DecisionTreeTrainer.DecisionTreeTrainerConfig

All Implemented Interfaces:

Serializable

Enclosing class:

DecisionTreeTrainer

public static class DecisionTreeTrainer.DecisionTreeTrainerConfig
extends SparkMLTrainer.MLTrainerConfig

Configuration for DecisionTreeTrainer.

See Also:

Serialized Form

Field Summary

Fields inherited from class co.cask.hydrator.plugin.spark.SparkMLTrainer.MLTrainerConfig

featureFieldsToExclude, featureFieldsToInclude, fileSetName, labelField, path

Constructor Summary

Constructors

Constructor and Description
DecisionTreeTrainer.DecisionTreeTrainerConfig()
DecisionTreeTrainer.DecisionTreeTrainerConfig(String fileSetName, String path, String featuresToInclude, String featuresToExclude, String cardinalityMapping, String labelField, Integer maxDepth, Integer maxBins)

Method Summary

Methods inherited from class co.cask.hydrator.plugin.spark.SparkMLTrainer.MLTrainerConfig

validate

Methods inherited from class co.cask.cdap.api.plugin.PluginConfig

containsMacro, getProperties

Methods inherited from class java.lang.Object

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

Constructor Detail

DecisionTreeTrainer.DecisionTreeTrainerConfig

public DecisionTreeTrainer.DecisionTreeTrainerConfig()

DecisionTreeTrainer.DecisionTreeTrainerConfig

public DecisionTreeTrainer.DecisionTreeTrainerConfig(String fileSetName,
                                             @Nullable
                                             String path,
                                             @Nullable
                                             String featuresToInclude,
                                             @Nullable
                                             String featuresToExclude,
                                             @Nullable
                                             String cardinalityMapping,
                                             String labelField,
                                             @Nullable
                                             Integer maxDepth,
                                             @Nullable
                                             Integer maxBins)