2. My first Model

Workflow: My first Model

This workflow reads the two datasets (training and test set) created in the workflow "1. Data Preparation". It then trains two machine learning algorithms to predict the income (target class = "income"; >50k or <=50k) based on the remaining input features of the adult dataset. In this workflow we use the following two algorithms:

1. Naive Bayes

2. Decision Tree (C4.5)

Lastly, both models are evaluated based on the test set (Scorer vs. Scorer (JavaScript) nodes) and the two mode performances are visualized and compared (ROC Curve and Decision Tree View nodes).

Training machine learning models

using training set and remaining input features of adult dataset

Applying trained models

on test set

Evaluating the trained models

Actual "income" value vs. predicted values

Visualization

ROC curves and decision tree

Reading data

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