CNTKLearner¶

class CNTKLearner.CNTKLearner(brainScript=None, dataFormat='text', dataTransfer='local', featureCount=1, featuresColumnName='features', gpuMachines=['127.0.0.1,1'], labelsColumnName='labels', localHdfsMount=None, parallelTrain=True, username='sshuser', weightPrecision='float', workingDir='tmp')[source]¶

Bases: mmlspark._CNTKLearner._CNTKLearner

Create CNTK model from existing java model

Parameters:	java_model (py4j.java_gateway.JavaObject) – see Scala CNTKModel documentation