def _bag_features(self, tree_num, input_data):
    split_data = array_ops.split(1, self.params.num_features, input_data)
    return array_ops.concat(1, [split_data[ind] for ind in self.params.bagged_features[tree_num]])
def _bag_features(self, tree_num, input_data):
    split_data = array_ops.split(1, self.params.num_features, input_data)
    return array_ops.concat(1, [split_data[ind] for ind in self.params.bagged_features[tree_num]])