File size: 1,776 Bytes
1f5470c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | import collections
import re
from keras.src.api_export import keras_export
from keras.src.backend.common import global_state
def auto_name(prefix):
prefix = to_snake_case(prefix)
return uniquify(prefix)
def uniquify(name):
object_name_uids = global_state.get_global_attribute(
"object_name_uids",
default=collections.defaultdict(int),
set_to_default=True,
)
if name in object_name_uids:
unique_name = f"{name}_{object_name_uids[name]}"
else:
unique_name = name
object_name_uids[name] += 1
return unique_name
def to_snake_case(name):
name = re.sub(r"\W+", "", name)
name = re.sub("(.)([A-Z][a-z]+)", r"\1_\2", name)
name = re.sub("([a-z])([A-Z])", r"\1_\2", name).lower()
return name
@keras_export("keras.backend.get_uid")
def get_uid(prefix=""):
"""Associates a string prefix with an integer counter.
Args:
prefix: String prefix to index.
Returns:
Unique integer ID.
Example:
>>> get_uid('dense')
1
>>> get_uid('dense')
2
"""
object_name_uids = global_state.get_global_attribute(
"object_name_uids",
default=collections.defaultdict(int),
set_to_default=True,
)
object_name_uids[prefix] += 1
return object_name_uids[prefix]
def reset_uids():
global_state.set_global_attribute(
"object_name_uids", collections.defaultdict(int)
)
def get_object_name(obj):
if hasattr(obj, "name"): # Most Keras objects.
return obj.name
elif hasattr(obj, "__name__"): # Function.
return to_snake_case(obj.__name__)
elif hasattr(obj, "__class__"): # Class instance.
return to_snake_case(obj.__class__.__name__)
return to_snake_case(str(obj))
|