Searched refs:ensemble_config (Results 1 – 3 of 3) sorted by relevance
54 stamp_token, ensemble_config = tree_ensemble_serialize(tree_ensemble_handle)62 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
920 const boosted_trees::trees::DecisionTreeEnsembleConfig& ensemble_config = in Compute() local944 ensemble_config.growing_metadata().num_trees_attempted(); in Compute()960 for (const auto& tree_metadata : ensemble_config.tree_metadata()) { in Compute()964 int tree_metadata_size = ensemble_config.tree_metadata_size(); in Compute()967 ? ensemble_config.tree_metadata(tree_metadata_size - 1) in Compute()971 ensemble_config.growing_metadata().num_layers_attempted(); in Compute()
890 ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()907 }""", ensemble_config)910 tree_ensemble_config=ensemble_config.SerializeToString(),944 ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()1010 }""", ensemble_config)1013 tree_ensemble_config=ensemble_config.SerializeToString(),1949 ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()1966 }""", ensemble_config)1969 tree_ensemble_config=ensemble_config.SerializeToString(),