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 training setwritten by workflowPrepare Datatest setwritten by workflowPrepare Dataattach class normalized distributionstraining to predict incomebased onremaining input featuresmale in bluefemale in redtraining to predict incomebased onremaining input featuresattach probabilitiesconfusion matrix + scoresconfusion matrix + scores File Reader File Reader Decision TreePredictor DecisionTree Learner ROC Curve Joiner Column Rename Column Rename Color Manager Decision Tree View Naive Bayes Learner Naive BayesPredictor Scorer (JavaScript) Scorer (JavaScript) 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 training setwritten by workflowPrepare Datatest setwritten by workflowPrepare Dataattach class normalized distributionstraining to predict incomebased onremaining input featuresmale in bluefemale in redtraining to predict incomebased onremaining input featuresattach probabilitiesconfusion matrix + scoresconfusion matrix + scores File Reader File Reader Decision TreePredictor DecisionTree Learner ROC Curve Joiner Column Rename Column Rename Color Manager Decision Tree View Naive Bayes Learner Naive BayesPredictor Scorer (JavaScript) Scorer (JavaScript)

Nodes

Extensions

Links