`MLaaS.ktrain`¶

File : ktrain.py Author : Valentin Kuznetsov Description: Keras based ML network to train over MNIST dataset

Module Contents¶

`modelGraph`	Provide input/output names used by TF Graph along with graph itself The code is based on TF saved_model_cli.py script.
`readData`	Helper function to read MNIST data and provide it to upstream code, e.g. to the training layer
`train`	train function for MNIST
`writer`	Writer provide write function for given model
`convert`	convert input model.pb into output model.pbtxt Based on internet search:
`main`	Main function

MLaaS.ktrain.modelGraph(model_dir)¶: Provide input/output names used by TF Graph along with graph itself The code is based on TF saved_model_cli.py script.

MLaaS.ktrain.readData(fin, num_classes)¶: Helper function to read MNIST data and provide it to upstream code, e.g. to the training layer

MLaaS.ktrain.train(fin, fout=None, model_name=None, epochs=1, batch_size=128, h5=False)¶: train function for MNIST

MLaaS.ktrain.writer(fout, model_name, model, input_shape, h5=False)¶: Writer provide write function for given model

MLaaS.ktrain.convert(fin, fout)¶

convert input model.pb into output model.pbtxt Based on internet search:

https://www.tensorflow.org/guide/saved_model
https://www.programcreek.com/python/example/123317/tensorflow.core.protobuf.saved_model_pb2.SavedModel

class MLaaS.ktrain.OptionParser¶

Initialization

User based option parser