Instance Property

modelParameters

The underlying parameters used when training the model.

Declaration

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.

struct MLDecisionTreeRegressor.ModelParameters

Parameters that affect the process of training a 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.