init(trainingData:targetColumn:featureColumns:parameters:)

Declaration

init(trainingData: DataFrame, targetColumn: String, featureColumns: [String]? = nil, parameters: MLDecisionTreeClassifier.ModelParameters = ModelParameters(validation: .split(strategy: .automatic))) throws

Parameters

• trainingData:

  The training data

• targetColumn:

  Name of the column containing the class labels

• featureColumns:

  Names of the columns containing feature values. If nil all columns, other than the target column, will be used as feature values.

• parameters:

  Model training parameters. See Modelparameters Swift.struct for the defaults.

See Also

Training a decision tree classifier asynchronously

• makeTrainingSession(trainingData:targetColumn:featureColumns:parameters:sessionParameters:)
• resume(_:)
• restoreTrainingSession(sessionParameters:)