AutoTrainedModel[TrainedModel <: Model[TrainedModel]] extends Model[TrainedModel] with ComplexParamsWritable with HasLabelCol with HasFeaturesCol
Defines common inheritance and functions across auto trained models.
AutoTrainer[TrainedModel <: Model[TrainedModel]] extends Estimator[TrainedModel] with HasLabelCol with ComplexParamsWritable with HasFeaturesCol with Wrappable
Defines common inheritance and parameters across trainers.
- trait CPISParams extends Wrappable with DefaultParamsWritable with HasLabelCol with HasScoresCol with HasScoredLabelsCol with HasScoredProbabilitiesCol with HasEvaluationMetric
ComputeModelStatistics extends Transformer with ComputeModelStatisticsParams with BasicLogging
Evaluates the given scored dataset.
- trait ComputeModelStatisticsParams extends Wrappable with DefaultParamsWritable with HasLabelCol with HasScoresCol with HasScoredLabelsCol with HasEvaluationMetric
ComputePerInstanceStatistics extends Transformer with CPISParams with BasicLogging
Evaluates the given scored dataset with per instance metrics.
MetricsLogger extends AnyRef
Helper class for logging metrics to log4j.
TrainClassifier extends Estimator[TrainedClassifierModel] with AutoTrainer[TrainedClassifierModel] with BasicLogging
Trains a classification model.
TrainRegressor extends Estimator[TrainedRegressorModel] with AutoTrainer[TrainedRegressorModel] with BasicLogging
Trains a regression model.
TrainedClassifierModel extends AutoTrainedModel[TrainedClassifierModel] with Wrappable with BasicLogging
Model produced by TrainClassifier.
TrainedRegressorModel extends AutoTrainedModel[TrainedRegressorModel] with Wrappable with BasicLogging
Model produced by TrainRegressor.
- object ComputeModelStatistics extends DefaultParamsReadable[ComputeModelStatistics] with Serializable
- object ComputePerInstanceStatistics extends DefaultParamsReadable[ComputePerInstanceStatistics] with Serializable
- object TrainClassifier extends ComplexParamsReadable[TrainClassifier] with Serializable
- object TrainRegressor extends ComplexParamsReadable[TrainRegressor] with Serializable
- object TrainedClassifierModel extends ComplexParamsReadable[TrainedClassifierModel] with Serializable
- object TrainedRegressorModel extends ComplexParamsReadable[TrainedRegressorModel] with Serializable