Icon

03_​Train_​MNIST_​classifier

Train MNIST classifier

This workflow trains a simple convolutional neural network (CNN) on the MNIST dataset via TensorFlow.

In order to run the example, please make sure you have the following KNIME extensions installed:

* KNIME Deep Learning - TensorFlow Integration (Labs)
* KNIME Image Processing (Community Contributions Trusted)
* KNIME Image Processing - Deep Learning Extension (Community Contributions Trusted)

You also need a local Python installation that includes TensorFlow. Please refer to https://www.knime.com/deeplearning/tensorflow for installation recommendations and further information.

Acknowledgements:

The architecture of the created network was taken but slightly changed from https://www.tensorflow.org/tutorials/layers.

The enclosed pictures are from the MNIST dataset (http://yann.lecun.com/exdb/mnist/) [1].

[1] Y. LeCun, L. Bottou, Y. Bengio, and P. Haffner. "Gradient-based learning applied to document recognition." Proceedings of the IEEE, 86(11):2278-2324, November 1998.

KNIME Deep Learning - TensorFlow - Train MNIST classifier This workflow trains a simple convolutional neural network (CNN) on the MNIST datasetvia TensorFlow. Create TensorFlowCNNTrain CNNExecute ontest dataSave the modelMNIST imagesMNIST imagestransform probabilitiesto predicted classes’ labelsopenView: Confusion Matrix DL PythonNetwork Creator DL PythonNetwork Learner DL Network Executor(deprecated) TensorFlowNetwork Writer Preparetraining data Prepare test data Format results Scorer Image Viewer KNIME Deep Learning - TensorFlow - Train MNIST classifier This workflow trains a simple convolutional neural network (CNN) on the MNIST datasetvia TensorFlow. Create TensorFlowCNNTrain CNNExecute ontest dataSave the modelMNIST imagesMNIST imagestransform probabilitiesto predicted classes’ labelsopenView: Confusion MatrixDL PythonNetwork Creator DL PythonNetwork Learner DL Network Executor(deprecated) TensorFlowNetwork Writer Preparetraining data Prepare test data Format results Scorer Image Viewer

Nodes

Extensions

Links

Download

To use this workflow in KNIME, download it from the below URL and open it in KNIME:

Download Workflow
Created by: bw
Created at: 2018-06-29
On NodePit since: 2019-07-12
Last update: 2024-04-18
Created with KNIME version: v4.1.0
Tags: deep learningTensorFlowimage classification

Deploy, schedule, execute, and monitor your KNIME workflows locally, in the cloud or on-premises – with our brand new NodePit Runner.

Try NodePit Runner!