def __init__(self, name, stamp_token=0, is_local=False, serialized_proto=''):
with ops.name_scope(name, 'TreeEnsemble') as name:
self._resource_handle = (
gen_boosted_trees_ops.boosted_trees_ensemble_resource_handle_op(
container='', shared_name=name, name=name))
create_op = gen_boosted_trees_ops.boosted_trees_create_ensemble(
self.resource_handle,
stamp_token,
tree_ensemble_serialized=serialized_proto)
is_initialized_op = (
gen_boosted_trees_ops.is_boosted_trees_ensemble_initialized(
self._resource_handle))
# Adds the variable to the savable list.
if not is_local:
saveable = _TreeEnsembleSavable(self.resource_handle, create_op,
self.resource_handle.name)
ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
resources.register_resource(
self.resource_handle,
create_op,
is_initialized_op,
is_shared=not is_local)
def _initialize(self):
return gen_boosted_trees_ops.boosted_trees_create_ensemble(
self.resource_handle,
self._stamp_token,
tree_ensemble_serialized=self._serialized_proto)
def initialize(self):
return gen_boosted_trees_ops.boosted_trees_create_ensemble(
self.resource_handle,
self._stamp_token,
tree_ensemble_serialized=self._serialized_proto)
