Logo image
Towards Interpretable Automated Machine Learning for STEM Career Prediction
Journal article   Open access   Peer reviewed

Towards Interpretable Automated Machine Learning for STEM Career Prediction

Ruitao Liu and Aixin Tan
Journal of educational data mining, Vol.12(2), pp.19-32
2020
DOI: 10.5281/zenodo.4008073
url
https://zenodo.org/record/4008073View
Open Access

Abstract

In this paper, we describe our solution to predict student STEM career choices during the 2017 ASSISTments Datamining Competition. We built a machine learning system that automatically reformats the data set, generates new features and prunes redundant ones, and performs model and feature selection. We designed the system to automatically find a model that optimizes prediction performance, yet the final model is a simple logistic regression that allows researchers to discover important features and study their effects on STEM career choices. We also compared our method to other methods, which revealed that the key to good prediction is proper feature enrichment in the beginning stage of the data analysis, while feature selection in a later stage allows a simpler final model.
Artificial Intelligence Automation Career Choice Careers Engineering Mathematics Middle School Students Prediction Science Careers Technical Occupations

Details

Metrics

7 Record Views
Logo image