WSO2 Machine Learner provides an interface to configure algorithms to build machine learning models using datasets. These models are used for tasks such as fraud detection, anomaly detection classification etc.
This guide walks you through the basic features of WSO2 ML to get you started. For this purpose, a dataset is analyzed and a ML model is trained to predict the possibility of a person suffering with breast cancer when data relating to a set of other bodily characteristics is provided.
The dataset used in this scenario contains 10 features to provide data on bodily characteristics, and a response variable with the following labels.
Label | Prediction |
---|---|
2 | This value indicates that the situation is benign. |
4 | This value indicates that the situation is malignant. |
The following steps demonstrate how the Machine Learner can use this dataset to make a prediction.
Before you begin,
- Install Oracle Java SE Development Kit (JDK) version 1.6.24 or later or 1.7.* and set the
JAVA_HOME
environment variable. - Download WSO2 ML.
- Start the ML by going to
<ML_HOME>/bin
using the command-line and executingwso2server.bat
(for Windows) orwso2server.sh
(for Linux.)
Step 1: Create a dataset
Follow the procedure below to upload the dataset based on which the training model will be created.
- Log into the ML UI using
admin
as both the username and password. The following will be displayed in the Home page.
- Click ADD DATASET to open the Create Dataset page.
In the Data Source field, click Choose File and browse for the
<ML_HOME>/samples/tuned/naive-bayes/breastCancerWisconsin.csv
file. Enter values for the rest of the parameters as shown below.Parameter Name Value Dataset Name Breast_Cancer_Dataset Version 1.0.0 Description Breast cancer data in Wisconsin. Source Type File Data Format CSV Column Header Available Yes - Click CREATE DATASET to save your changes. The Datasets page will open and the dataset you entered will be displayed as follows.
Note that the status of the dataset is Processing. - Click REFRESH. The status of the dataset will change to Processed as shown below.
Step 2: Create a project
Follow the procedure below to create a project for the dataset uploaded.
- Log into the ML Management Console if you are not already logged in.
- Click ADD PROJECT.
If you are already logged in, you can click CREATE PROJECT in the DATASETS page as shown below.
In the Create Project page, enter information as shown below.
Parameter Name Description Project Name Breast_Cancer_data_analytics_project Description This project performs predictive analysis on the breast cancer data in Wisconsin. Dataset Breast_Cancer_Dataset - Click Create Project to save the information. The project will be displayed in the Projects page as follows.
Step 3: Create an analysis and train a model
Follow the procedure below to analyse the Breast_Cancer_Dataset
dataset, and then create a training model based on that analysis.
- Log into the ML UI if you are not already logged in.
- Click the You have X projects link as shown below.
- Click on the Breast_Cancer_data_analytics_project project to expand it.
- Enter breast_cancer_analysis_1 as the analysis name and click CREATE ANALYSIS. The following page will appear displaying the summary statistics.
- Click Next without making any changes to the summary statistics.
The Explore view will open. You will notice that Parallel Sets and Trellis Chart visualisations are enabled, and Scatter Plot and Cluster Diagram visualisations are disabled. This is determined by the feature types of the dataset. Select and clear the checkboxes for categorical features as follows.
Click Next. The Algorithms view will be displayed. Enter values as shown below.
Parameter Value Algorithm name LOGISTIC REGRESSION L_BFGS Response variable Class Train data fraction 0.7 - Click Next. The Parameters view will appear. Enter L2 as the reg type.
- Click Next. The Model view will appear. Select Breast_Cancer_Dataset-1.0.0 as the dataset version.
- Click RUN to train the model.
The analysis will be created and displayed for the project as shown below.
Step 4: Predict using the model
Follow the procedure below to make a prediction based on the training model you created.
- Log into the ML UI if you are not already logged in.
- Click the You have X projects link as shown below to open the Projects window.
- Click MODELS for the breast_cancer_analysis_1 analysis.
- Click Predict on the model displayed.
Enter values in the Predict page as shown below.
Parameter Name Value Prediction Source Feature values SampleCodeNumber 1018561 ClumpThickness 2 UniformityOfCellSize 1 UniformityOfCellShape 1 MarginalAdhesion 1 SingleEpithelialCellSize 2 BareNuclei 1 BlandChromati 1 NormalNucleoli 1 Mitoses 5 - Click Predict. The prediction will be displayed as follows.