Contents

outputName

The name of the output from a feature extraction layer within the model.

Declaration

var outputName: String?

Discussion

When training an image classifier with a custom feature extractor, use outputName to select a layer within the model that has an output of MLMultiArray.

Set outputName to nil if the model (specified by modelPath) has only one output of type MLMultiArray.

See Also

Configuring a custom feature extractor