Contents

init(trainingData:textColumn:labelColumn:parameters:)

Creates a text classifier.

Declaration

init(trainingData: DataFrame, textColumn: String, labelColumn: String, parameters: MLTextClassifier.ModelParameters = ModelParameters(validation: .split(strategy: .automatic))) throws

Parameters

  • trainingData:

    A data frame specifying the training data.

  • textColumn:

    The name of the text column in the provided trainingData.

  • labelColumn:

    The name of the label column in the provided trainingData.

  • parameters:

    Model training parameters.

See Also

Creating and training a text classifier