Z.H. X. comments

Repositories
Issues
Comments

Results 3 comments of


                                            Z.H. X.

关于split_gain的问题

> 您好： > 我在阅读您的代码的时候发现一个问题，self.gain = getItemByTree(self, 'split_gain')，这行代码应该是获取节点每次分裂的信息增益，但是在getItemByTree里面的getFeature里面并没有相对应的操作。 > def getItemByTree(tree, item='split_feature'): > root = tree.raw['tree_structure'] > split_nodes = tree.split_nodes > res = np.zeros(split_nodes+tree.raw['num_leaves'], dtype=np.int32) > if 'value' in item or...

max_ntree_per_split is 0

Thanks, there may be some corner cases that make it an error, please consider fixing it in your code.

What is the feature_sizes exactly?

please see https://github.com/motefly/DeepGBM/blob/master/preprocess/encoding_cate.py#L48 .