Generating a Model Using the Decision Tree Algorithm

Introduction

This sample demonstrates how a model is generated out of a data set using the decision tree algorithm. The sample uses a data set to generate a model, which is divided into two sets for training and testing.

Prerequisites

Follow the steps below to set up the prerequisites before you start.

1. Download WSO2 Machine Learner, and start the server. For information on setting up and running WSO2 ML, see Getting Started.
2. Download and install  jq (CLI JSON processor). For instructions, see jq Documentation.

Executing the sample

You can execute the sample either using the CLI or using the ML UI as described below.

Using the CLI

Follow the steps below to execute the sample using the CLI.

1. Navigate to <ML_HOME>/samples/default/decision-tree/ directory using the CLI.

   <ML_HOME> refers to the downloaded product-ml directory with the source code of the product

2. Execute the following command to execute the sample: ./model-generation.sh

Output of the sample

 Once the sample is successfully executed, you can obtain the following output.

1. By default, the sample generates the model in the <ML_HOME>/models/  directory of your machine. For example, the generated file is in the following format denoting the date and time when it was generated:  wso2-ml-decision-tree-sample-analysis.Model.2015-09-03_11-01-46

2. The sample executes the generated model on the  <ML_HOME>/samples/default/decision-tree/prediction-test data set, and it prints the value ["1"] as the prediction result In the CLI logs.