ImageFeaturizer

class ImageFeaturizer.ImageFeaturizer(cntkModel=None, cutOutputLayers=1, dropNa=True, inputCol=None, layerNames=None, outputCol=None)[source]

Bases: mmlspark._ImageFeaturizer._ImageFeaturizer

Parameters:
  • SparkSession (SparkSession) – The SparkSession that will be used to find the model
  • ocation (str) – The location of the model, either on local or HDFS
setMiniBatchSize(size)[source]
setModel(modelSchema)[source]
setModelLocation(location)[source]