Skip to content

jihli/Resident-Salary-Prediction

Repository files navigation

Resident Salary Prediction:

Part 1 Data Exploration: Statistical analysis/Exploratory Data Analysis

Part 2 Feature Preprocessing: Data Spliting /One Hot Encoding/Data Standardization

Part 3 Model Training and Evaluation: Models Comparison/Model Evaluation - Confusion Matrix/ROC & AUC

Part 4 Find the Best Machine Learning Model: Optimizing Hyperparameters of LogisticRegression&KNN&Random Forest via Grid Search and pick up the best one

Part 5 Model Extra Functionality - Importance Feature: L1&L2 regularization to logistic regression/Random Forest - Feature Importance

----About Dataset----

Extraction was done by Barry Becker from the 1994 Census database. Prediction task is to determine whether a person makes over 50K a year.

Columns are: age: continuous.

workclass: Private, Self-emp-not-inc, Self-emp-inc, Federal-gov, Local-gov, State-gov, Without-pay, Never-worked.

fnlwgt: continuous.

education: Bachelors, Some-college, 11th, HS-grad, Prof-school, Assoc-acdm, Assoc-voc, 9th, 7th-8th, 12th, Masters, 1st-4th, 10th, Doctorate, 5th-6th, Preschool. education-num: continuous.

marital-status: Married-civ-spouse, Divorced, Never-married, Separated, Widowed, Married-spouse-absent, Married-AF-spouse.

occupation: Tech-support, Craft-repair, Other-service, Sales, Exec-managerial, Prof-specialty, Handlers-cleaners, Machine-op-inspct, Adm-clerical, Farming-fishing, Transport-moving, Priv-house-serv, Protective-serv, Armed-Forces.

relationship: Wife, Own-child, Husband, Not-in-family, Other-relative, Unmarried.

race: White, Asian-Pac-Islander, Amer-Indian-Eskimo, Other, Black.

sex: Female, Male.

capital-gain: continuous.

capital-loss: continuous.

hours-per-week: continuous.

native-country: United-States, Cambodia, England, Puerto-Rico, Canada, Germany, Outlying-US(Guam-USVI-etc), India, Japan, Greece, South, China, Cuba, Iran, Honduras, Philippines, Italy, Poland, Jamaica, Vietnam, Mexico, Portugal, Ireland, France, Dominican-Republic, Laos, Ecuador, Taiwan, Haiti, Columbia, Hungary, Guatemala, Nicaragua, Scotland, Thailand, Yugoslavia, El-Salvador, Trinadad&Tobago, Peru, Hong, Holand-Netherlands.

salary: <=50K or >50K

Source: https://archive.ics.uci.edu/ml/datasets/Census+Income

Thumbnail: https://www.deskera.com/blog/net-salary/