CNTKModel¶
-
class
CNTKModel.
CNTKModel
(batchInput=True, convertOutputToDenseVector=True, feedDict={'ARGUMENT_0': 'ARGUMENT_0'}, fetchDict={'OUTPUT_0': 'OUTPUT_0'}, miniBatcher=None, model=None, shapeOutput=False)[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
-
setFeedDict
(dict)[source]¶ Parameters: feedDict (dict) – Map of CNTK Variable names (keys) and Column Names (values) (default: Map(ARGUMENT_0 -> ARGUMENT_0))