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 __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 is_initialized(self): return gen_boosted_trees_ops.is_boosted_trees_ensemble_initialized( self.resource_handle)
def is_initialized(self): return gen_boosted_trees_ops.is_boosted_trees_ensemble_initialized( self.resource_handle)