pip install pandas
pip install scikit-learn
A gender classification code in python to predict the gender based on height weight and shoe size
The library used is sklearn or scikit learn
Used four different classifiers for this purpose:
- Decision Tree Classifier
- Random Forest Classifier
- Logistic Regression
- SVC (Support Vector Classifier)
Also the most accurate of prediction is also calculated using package numpy
Now we can execute the program as
test_data = [[190, 70, 43],[154, 75, 38],[181,65,40]] test_labels = [‘male’,’female’,’male’]
Output:
[‘male’ ‘female’ ‘female’]
[‘male’ ‘female’ ‘female’]
[‘female’ ‘female’ ‘female’]
[‘female’ ‘female’ ‘female’]
Decision Tree is the best classifier for this problem
-
- Apprentissage du modèle
-
- Test des modèles (evaluer les modèles)
-
- Validation du modèle (choisir le bon modèle) et enrichir la data
-
- Generer le modele
-
- Utiliser le modele