https://github.com/scikit-learn/scikit-learn
hello guys,
In this tutorial we will learn about classification of gender (male or female) if given the body metrics of the subject and predict whether it's male or female after training the classifier over an sample data of body metrics labelled male and female.
Choosing best classifier for an problem can decided based upon their results and requirement check out this link to know more.
If you haven't installed Scikit-Learn and you want to do it then you can check out this link, it has all the information regarding installing the Scikit-Learn library . First create a directory in which you can run tests and predict using coded Python file. The first step will obviously be opening the code editor. After opening the code editor you will need to import required module from sklearn(package) which you can do by typing from sklearn import (classifier name).
So, now we're ready to write our script we'll start by importing it first as we should for all the dependencies we're going to use a specific sub module of scikit-learn(sklearn) called tree that wil let us build a machine learning model called a decision tree.
It is like a flowchart that stores data , it asks each labelled data point it recieves a yes-or-no question , does it contain X or not if the answer is yes the data moves one direction if the answer is no it moves in the other it'll build every node in the tree the more data points it receives then when we have a new unlabeled data point we can feed it to the tree it'll ask if a series of questions until it labels it that label is our classification.
The more data we train it on the more accurate the classification.
Lets start by creating our data set programmatically we'll write our first variable for dataset , X as a list of Lists a variable is a value that can change and we'll store a list of Lists in it a list is a data type in python that can store a sequence of values here each value is a list itself that contains three numbers that represent the height , weight and shoe-size of a person we'll write 11 of these so our data set size is only 11 people and we will write one more variable called Y to store list of labels each label is a gender(male or female) and is associated with a list of body metrics in the previous list , we'll write them as strings which is a data type used to represent text instead of numbers.
Now that we have our data set we'll want to define a variable to store our decision tree model let's call it clf.
CLF short for classifier and it'll store our decision tree classifier we can reference our tree dependency directly by calling it here and initialize the decision tree by calling the decision tree method on the tree object.
Now that we have our tree variable we can train it on our data set we'll call the fit method on the classifier variable which takes two arguments will store our X and Y variables as the arguments and the result will be stored in the updated clf variable the fit method trains the decision tree on our data set.
Let's test it by classifying the gender of someone given a new list of body metrics we'll create a variable called prediction to store the result and call the predict method of our decision tree to predict the gender given these three values in a list.
then we can print it out determine all via the print command we can run the script in terminal by saving it as filename . py and running it via the Python filename.py command.
Outputs:-
Model Name:- Decision Tree Classifier
| Input Metrics (height , weight and shoe-size) | Output Label |
|---|---|
| 140, 60, 33 | Female |
| 160, 80 ,43 | Male |
So, in the above data set metrics shoe-size metric can also be replaced by waist size because to classify an subject from other we need an discriminating factor between them , so the more discriminating factor with more data values provides accurate prediction.
You can also use other classifiers to make classification models for prediction I've given the code for those too so try to compare the results and you can even change the metrics factors too.
