CNTKModel¶

class CNTKModel.CNTKModel(inputCol=None, inputNode=0, miniBatcher=None, model=None, outputCol=None, outputNodeIndex=None, outputNodeName=None)[source]¶

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

rebroadcastCNTKModel(sparkSession)[source]¶

setMiniBatchSize(n)[source]¶

setModelLocation(sparkSession, location)[source]¶