Searched refs:decision_tree_ (Results 1 – 4 of 4) sorted by relevance
23 return *decision_tree_; in decision_tree()27 return decision_tree_->nodes_size(); in get_size()31 : decision_tree_( in TensorForestTreeResource()36 return decision_tree_->nodes(id).leaf().vector().value(dimension_id); in get_prediction()46 const Node& current = decision_tree_->nodes(current_id); in TraverseTree()62 return ParseProtoUnlimited(decision_tree_, serialized); in InitFromSerialized()68 decision_tree_ = protobuf::Arena::CreateMessage<boosted_trees::Tree>(&arena_); in Reset()
61 boosted_trees::Tree* decision_tree_; variable
37 decision_tree_->decision_tree().nodes_size(), "]"); in DebugString()42 const decision_trees::Model& decision_tree() const { return *decision_tree_; } in decision_tree()45 return decision_tree_.get(); in mutable_decision_tree()49 return decision_tree_->decision_tree().nodes(id).leaf(); in get_leaf()53 return decision_tree_->mutable_decision_tree()->mutable_nodes(id); in get_mutable_tree_node()58 void Reset() { decision_tree_.reset(new decision_trees::Model()); } in Reset()76 std::unique_ptr<decision_trees::Model> decision_tree_; variable
25 : params_(params), decision_tree_(new decision_trees::Model()) { in DecisionTreeResource()32 const DecisionTree& tree = decision_tree_->decision_tree(); in TraverseTree()55 DecisionTree* tree = decision_tree_->mutable_decision_tree(); in SplitNode()84 DecisionTree* tree = decision_tree_->mutable_decision_tree(); in MaybeInitialize()