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

Download WSO2 Machine Learner, and start the server. 

Building the sample

Execute the following command to download the source code of the product: git clone https://github.com/wso2/product-ml.git

Executing the sample

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

1. If you already executed a sample before, execute the following command to remove the databases created: rm -rf repository/database/WSO2ML_DB.*

2. Navigate to <ML_HOME>/samples/rest-api/decision-tree/ directory using the CLI.

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

3. Execute the following command to execute the sample: sh 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 /tmp/ directory of your machine. For example, the generated file is in the following format denoting the date and time when it was generated: model.1.2015-03-20_13:29:10

   You can change the location where the output model file is saved, by defining the folder path of the required location for the value of the location element in the <ML_HOME>/samples/rest-api/decision-tree/create-model-storage file as shown below.

   {
   "type" : "file",
   "location" : "/tmp"
   }

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