公开(公告)号：US20200311559A1

公开(公告)日：2020-10-01

申请号：US16902063

申请日：2020-06-15

申请人： Rita Chattopadhyay ,  Rajesh Bansal ,  Yuming Ma ,  Mrittika Ganguli

发明人： Rita Chattopadhyay ,  Rajesh Bansal ,  Yuming Ma ,  Mrittika Ganguli

IPC分类号： G06N5/00 ,  G06N5/04 ,  G06N20/00

摘要： In one embodiment, an edge computing device for performing decision tree training and inference includes interface circuitry and processing circuitry. The interface circuitry receives training data and inference data that is captured, at least partially, by sensor(s). The training data corresponds to a plurality of labeled instances of a feature set, and the inference data corresponds to an unlabeled instance of the feature set. The processing circuitry: computes a set of feature value checkpoints that indicate, for each feature of the feature set, a subset of potential feature values to be evaluated for splitting tree nodes of a decision tree model; trains the decision tree model based on the training data and the set of feature value checkpoints; and performs inference using the decision tree model to predict a target variable for the unlabeled instance of the feature set.