https://github.com/redtachyon/sklearn-modelling
Predictive modelling of the Adult Data Set (https://archive.ics.uci.edu/ml/datasets/adult)
https://github.com/redtachyon/sklearn-modelling
Last synced: over 1 year ago
JSON representation
Predictive modelling of the Adult Data Set (https://archive.ics.uci.edu/ml/datasets/adult)
- Host: GitHub
- URL: https://github.com/redtachyon/sklearn-modelling
- Owner: RedTachyon
- Created: 2017-08-27T12:08:24.000Z (almost 9 years ago)
- Default Branch: master
- Last Pushed: 2017-08-29T21:47:44.000Z (almost 9 years ago)
- Last Synced: 2025-01-24T12:13:15.746Z (over 1 year ago)
- Language: Python
- Homepage:
- Size: 641 KB
- Stars: 0
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# yazabi-sklearn-project
This repo contains two modules detailed in the assignment: data\_preprocessing.py and train\_and\_test.py, and several convenience modules for training and evaluating models.
Important functions:
- get\_all\_data from train\_and\_test.py allows you to extract all data as a feature matrices.
- run\_search from grid\_search.py allows you to do grid search on a model.
- train\_and\_validate from train\_and\_test.py allows you to train a model and evaluate its performance on the test set
eval\_model.py contains a convenient CLI interface for calling train\_and\_validate. Use it like this: ./eval\_model.py svm -n
Currently supported models:
1. Gaussian Naive Bayes (naive-bayes and naive-bayes-g)
2. Multinomial Naive Bayes (naive-bayes-m) - doesn't support normalized features
3. Bernoulli Naive Bayes (naive-bayes-b)
4. Decision Tree (decision-tree)
5. K Nearest Neighbors (knn)
6. Linear Support Vector Machine (svm)
7. Random Forest (rf)
8. Extra Trees (et)
9. Logistic Regression (logreg)