Workflow: Predicting a binary variable with the Decision Tree block
Ian Balanzá-Davis
Altair Employee
The Decision Tree block enables you to apply a Decision Tree predictive model to an input dataset.
The following demonstrates how to use the Decision Tree block to predict a dependent Score variable from an input dataset basketball_shots.csv (containing observations that detail a basketball shot in a professional game and the person taking the shot) based on other independent variables dataset:
- Import the basketball_shots.csv dataset onto a Workflow canvas using the Text File Import block.
- Expand the Model Training group in the Workflow palette, then click and drag a Decision Tree block onto the Workflow canvas.
- Click the Output port of the basketball_shots dataset block and drag a connection towards the Input port of the Decision Tree block.
- Double-click the Decision Tree block to display the Decision Tree Editor view and the Decision Tree Preferences dialog box.
- In the Decision Tree Preferences dialog box:
- In the Dependent variable drop-down list, select Score.
- In the Target Category drop-down list, select 1 (one).
- In the Unselected Independent Variables list, press and hold CTRL and select angle, distance_feet, height, position, and weight.
- Click Select to move the specified variables to the Selected Independent Variables list.
- Click OK to save the configuration and close the Decision Tree Preferences dialog box.
- Right click the 0:Score node and select Grow (C4.5) to train the model.
- Close the Decision Tree Editor View and save the configuration when prompted.
A green execution status is displayed in the Output port of the Decision Tree block. The Decision Tree block output can be used with a Score Block in order to make predictions on a dataset.
0