System for automatic, simultaneous feature selection and hyperparameter tuning for a machine learning model

    公开(公告)号:US10600005B2

    公开(公告)日:2020-03-24

    申请号:US16411590

    申请日:2019-05-14

    IPC分类号: G06N20/00 G06N3/08

    摘要: A computing device selects a feature set and hyperparameters for a machine learning model to predict a value for a characteristic in a scoring dataset. A number of training model iterations is determined. A unique evaluation pair is selected for each iteration that indicates a feature set selected from feature sets and a hyperparameter configuration selected from hyperparameter configurations. A machine learning model is trained using each unique evaluation pair. Each trained machine learning model is validated to compute a performance measure value. An estimation model is trained with the feature set, the hyperparameter configuration, and the performance measure value computed for unique evaluation pair. The trained estimation model is executed to compute the performance measure value for each unique evaluation pair. A final feature set and a final hyperparameter configuration are selected based on the computed performance measure value.

    SYSTEM FOR AUTOMATIC, SIMULTANEOUS FEATURE SELECTION AND HYPERPARAMETER TUNING FOR A MACHINE LEARNING MODEL

    公开(公告)号:US20190370684A1

    公开(公告)日:2019-12-05

    申请号:US16411590

    申请日:2019-05-14

    IPC分类号: G06N20/00

    摘要: A computing device selects a feature set and hyperparameters for a machine learning model to predict a value for a characteristic in a scoring dataset. A number of training model iterations is determined. A unique evaluation pair is selected for each iteration that indicates a feature set selected from feature sets and a hyperparameter configuration selected from hyperparameter configurations. A machine learning model is trained using each unique evaluation pair. Each trained machine learning model is validated to compute a performance measure value. An estimation model is trained with the feature set, the hyperparameter configuration, and the performance measure value computed for unique evaluation pair. The trained estimation model is executed to compute the performance measure value for each unique evaluation pair. A final feature set and a final hyperparameter configuration are selected based on the computed performance measure value.