Structure

MLDecisionTreeRegressor.ModelParameters

Parameters that affect the process of training a model.

Declaration

struct MLDecisionTreeRegressor.ModelParameters

Topics

Describing Parameters

var description: String

A text representation of the model parameters for a decision tree regressor.

var debugDescription: String

A text representation of the model parameters for a decision tree regressor that’s suitable for output during debugging.

var playgroundDescription: Any

A text representation of the model parameters for a decision tree regressor.

See Also

Creating and Training a Decision Tree Regressor

init(trainingData: MLDataTable, targetColumn: String, featureColumns: [String]?, parameters: MLDecisionTreeRegressor.ModelParameters)

Creates a Decision Tree Regressor from the feature columns in the training data to predict the values in the target column.

let modelParameters: MLDecisionTreeRegressor.ModelParameters

The underlying parameters used when training the model.

var targetColumn: String

The name of the column you selected at initialization to define which feature the regressor predicts.

var featureColumns: [String]

The names of the columns you selected at initialization to train the regressor.