Icon

2. My First Model

KBL: My First Classification Model
Workflow: My First Model This workflow reads the data sets written by the Prepare Data workflow. It then trains two machine learning algorithms:- Naive Bayes- Decision Tree (C4.5)to predict Income (>50k / <=50K) based on the remaining input features.Finally, it scores both models on the test set.The last section is about visualization: ROC Curves and decision tree visualization. Training MachineLearning Models ApplyingModels Scoring predicted values vs.original values Visualization ROC curves anddecision tree attach class normalized distributionstraining_set.csv filewritten by workflowPrepare Datatraining to predict incomebased onremaining input featurestraining to predict incomebased onremaining input featuresattach probabilitiesconfusion matrix + scoresconfusion matrix + scorestest_set.csv filewritten by workflowPrepare Data Decision TreePredictor CSV Reader DecisionTree Learner ROC Curve Joiner Column Rename Column Rename Decision Tree View Naive Bayes Learner Naive BayesPredictor Scorer (JavaScript) Scorer (JavaScript) CSV Reader Workflow: My First Model This workflow reads the data sets written by the Prepare Data workflow. It then trains two machine learning algorithms:- Naive Bayes- Decision Tree (C4.5)to predict Income (>50k / <=50K) based on the remaining input features.Finally, it scores both models on the test set.The last section is about visualization: ROC Curves and decision tree visualization. Training MachineLearning Models ApplyingModels Scoring predicted values vs.original values Visualization ROC curves anddecision tree attach class normalized distributionstraining_set.csv filewritten by workflowPrepare Datatraining to predict incomebased onremaining input featurestraining to predict incomebased onremaining input featuresattach probabilitiesconfusion matrix + scoresconfusion matrix + scorestest_set.csv filewritten by workflowPrepare Data Decision TreePredictor CSV Reader DecisionTree Learner ROC Curve Joiner Column Rename Column Rename Decision Tree View Naive Bayes Learner Naive BayesPredictor Scorer (JavaScript) Scorer (JavaScript) CSV Reader

Nodes

Extensions

Links