Icon

3. My first ANN

<p><strong>My first Artificial Neural Network</strong></p><p>This workflow reads the two datasets (training and test set) created in the workflow "1. Data Preparation" and further preprocesses the data. It then trains a Multilayer Perceptron (MLP), a feedforward artificial neural network, to predict the income (target class = "income"; &gt;50k or &lt;=50k) based on the remaining input features of the adult dataset.</p><p>Lastly, the trained ANN is evaluated based on the test set (<em>ROC Curve </em>node) and saved in the KBLData folder (<em>Model Writer</em> node).</p>

URL: KNIME Beginner's Luck (Book Homepage) https://www.knime.com/knimepress/beginners-luck

Workflow: My first Artificial Neural Network


This workflow reads the two datasets (training and test set) created in the workflow "1. Data Preparation" and further preprocesses the data. It then trains a Multilayer Perceptron (MLP), a feedforward artificial neural network, to predict the income (target class = "income"; >50k or <=50k) based on the remaining input features of the adult dataset.

Lastly, the trained ANN is evaluated based on the test set (ROC Curve node) and saved in the KBLData folder (Model Writer node).

Preprocessing

  • Filter data: Keep only number columns & "income" (target column)

  • Normalize data: Min-Max-normalization

Reading data

Training and test set created in workflow "1. Data Preparation".

Training ANN

Visualization

ROC curves and decision tree

Applying trained model

Save model

Save trained ANN to KBLdata folder

adult_training_set.csv
CSV Reader
adult_test_set.csv
CSV Reader
save model
Model Writer
Convert "sex"from string to integer (0/1)
Expression
apply MLP
MultiLayerPerceptron Predictor
Min-Max-Normalizationin [0,1]
Normalizer
draw ROCcurve
ROC Curve
train MLP architecture 8-4-1
RProp MLP Learner
Convert "sex"from string to integer (0/1)
Expression
Keep only numbersand "income"
Column Filter
Keep only numbersand "income"
Column Filter
ApplyMin-Max-Normalization
Normalizer (Apply)

Nodes

Extensions

Links