Sayali Joshi, Keng-Ming Lin, Antonio Bing
2024-10-30
To determine key predictors that influence the number of employees in firms as per the ‘Census’ dataset.
Data Summary:
Dataset Details:
Feature Importance Analysis Using H2O: “Leveraged H2O’s automated machine learning framework to compute variable importance, identifying the most influential predictors for employee count.”
Key Insights from H2O Analysis: “H2O’s feature importance analysis highlighted [mention top 2-3 variables, e.g., ‘industry type,’ ‘firm size,’ and ‘region’] as the most impactful factors influencing workforce size.”
Graph 1 – Actual vs. Predicted Values
Graph 2 – Residual Plot
1> The scatter plot (Graph 1) reveals that the model performs well for smaller firms, with actual values closely aligning with predicted values. However, variance increases for larger firms, suggesting potential room for improvement in handling outliers or extreme values.
2> The residual plot (Graph 2) highlights issues with heteroscedasticity, where prediction errors are more pronounced for firms with higher employee counts. This suggests the need for additional feature engineering or the application of advanced algorithms to address variability in large firms. Insights from Variable Importance:
3> The analysis shows that “Period” and specific industry categories (e.g., NAICS2017.23 for Construction) play a significant role in predicting employer firm counts, indicating temporal trends and industry-specific growth patterns are key factors.
4> Demographic variables (e.g., ownership characteristics like gender or ethnicity) show lower importance in the model, suggesting that industry and time-based factors dominate the prediction landscape.
Handling Skew: The presence of highly skewed industries in Graph 1 and the residual dispersion in Graph 2 indicates the need for better handling of non-normal data distributions.
Model Refinement: Incorporating interaction terms or non-linear models could improve accuracy for larger firms, as their prediction errors disproportionately affect model performance. Practical Implications:
Policymakers and businesses should focus on high-impact variables like industry type and temporal changes to understand business growth patterns better.
Supporting small firms in skewed industries may help balance disparities, as indicated by the residual and variable importance analysis.
Thank You..