<s> from container import * <EOL> class GraphModel ( Container ) : <EOL> def __init__ ( self , <EOL> stages , <EOL> outputStageNames , <EOL> costFn , <EOL> inputDim = <NUM_LIT:0> , <EOL> outputDim = <NUM_LIT:0> , <EOL> name = None , <EOL> decisionFn = None , <EOL> specFilename = None ) : <EOL> Container . __init__ ( self , <EOL> name = name , <EOL> stages = stages , <EOL> inputNames = [ '<STR_LIT:input>' ] , <EOL> outputStageNames = outputStageNames , <EOL> inputDim = inputDim , <EOL> outputDim = outputDim , <EOL> outputdEdX = True ) <EOL> self . getCost = costFn <EOL> self . predict = decisionFn <EOL> self . specFilename = specFilename </s>
