CNTKModel

class CNTKModel.CNTKModel(batchInput=True, convertOutputToDenseVector=True, feedDict={'argument_0': 'argument_0'}, fetchDict={'output_0': 'output_0'}, miniBatcher=None, model=None, shapeOutput=False)

Bases: mmlspark._CNTKModel._CNTKModel

Parameters:

• SparkSession (SparkSession) – The SparkSession that will be used to find the model
• location (str) – The location of the model, either on local or HDFS

getInputCol()

getInputNode()

getInputNodeIndex()

getInputShapes()

getOutputCol()

getOutputNode()

getOutputNodeIndex()

rebroadcastCNTKModel(sparkSession)

setInputCol(n)

setInputNode(n)

setInputNodeIndex(n)

setMiniBatchSize(n)

setModelLocation(location)

setOutputCol(n)

setOutputNode(n)

setOutputNodeIndex(n)