We had a FY 2019 H-1b Visa Data Distribution post, now we are having models to predict the annual wage and wage level using the same dataset.

Buisiness question: Foreign worker with specialized knowledge, what would the annual wage look like?

Predict annual wage

The list of columns:

['Unnamed: 0',
'VISA_CLASS',
'CASE_STATUS',
'SOC_TITLE',
'SOC_CODE',
'NAICS_CODE',
'EMPLOYER_NAME',
'AGENT_ATTORNEY_LAW_FIRM',
'WORKSITE_ADDRESS_CLEAN',
'WORKSITE_CITY_CLEAN',
'WORKSITE_STATE',
'WORKSITE_POSTAL_CODE',
'WAGE_RATE_ANNUAL',
'PREVAILING_WAGE_RATE_ANNUAL',
'PW_WAGE_LEVEL',
'PW_OES_YEAR',
'GPS']

First model: Linear Regression model with feature PREVAILING_WAGE_RATE_ANNUAL

y = 28099.74951382118 + 0.8969421177290609x1
R^2: 0.5468001738806643

improvement: using gradient boosting


The H-1B is a visa in the United States which allows U.S. employers to employ foreign workers in specialty occupations.

H-1B Visa Wikipedia page, https://en.wikipedia.org/wiki/H-1B_visa

Dataset: Datasets of this year and previous years can be found in https://www.dol.gov/agencies/eta/foreign-labor/performance, I use the H-1B FY2019.xlsx dataset.

The key questions I will be covering include:

  1. What is the number of h-1b visa — annual salary distribution look like?
Number of h-1b visa — Annual salary distribution

Observation: The majority annual salary for 2019 h-1b visa petition lies in 75k~100k.

2. What are the top 50 job titles in the h-1b petition?

Yuhan Wang

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store