2021/03/03 by Cerulli, Giovanni
#Computation (stat.CO) #FOS: Computer and information sciences #Machine Learning (cs.LG) #Mathematical Software (cs.MS)
paper · doi:10.48550/arxiv.2103.03122
We present two related Stata modules, rmlstata and cmlstata, for fitting popular Machine Learning (ML) methods both in regression and classification settings. Using the recent Stata/Python integration platform (sfi) of Stata 16, these commands provide hyper-parameters' optimal tuning via K-fold cross-validation using greed search. More specifically, they make use of the Python Scikit-learn API to carry out both cross-validation and outcome/label prediction.