easy-cs-rec-custommodel 0.8.6__py2.py3-none-any.whl

easy_rec/python/protos/uniter_pb2.py

@@ -0,0 +1,307 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: easy_rec/python/protos/uniter.proto
+
+import sys
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pb2
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+
+from easy_rec.python.protos import dnn_pb2 as easy__rec_dot_python_dot_protos_dot_dnn__pb2  # NOQA
+
+_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+    name='easy_rec/python/protos/uniter.proto',
+    package='protos',
+    syntax='proto2',
+    serialized_pb=_b(
+        '\n#easy_rec/python/protos/uniter.proto\x12\x06protos\x1a easy_rec/python/protos/dnn.proto\"\xf2\x02\n\x0bUniterTower\x12\x13\n\x0bhidden_size\x18\x01 \x02(\r\x12\x19\n\x11num_hidden_layers\x18\x02 \x02(\r\x12\x1b\n\x13num_attention_heads\x18\x03 \x02(\r\x12\x19\n\x11intermediate_size\x18\x04 \x02(\r\x12\x18\n\nhidden_act\x18\x05 \x02(\t:\x04gelu\x12 \n\x13hidden_dropout_prob\x18\x06 \x02(\x02:\x03\x30.1\x12)\n\x1c\x61ttention_probs_dropout_prob\x18\x07 \x02(\x02:\x03\x30.1\x12$\n\x17max_position_embeddings\x18\x08 \x02(\r:\x03\x35\x31\x32\x12%\n\x17use_position_embeddings\x18\t \x02(\x08:\x04true\x12\x1f\n\x11initializer_range\x18\n \x02(\x02:\x04\x30.02\x12&\n\x11other_feature_dnn\x18\x0b \x01(\x0b\x32\x0b.protos.DNN\"M\n\x06Uniter\x12#\n\x06\x63onfig\x18\x01 \x02(\x0b\x32\x13.protos.UniterTower\x12\x1e\n\tfinal_dnn\x18\x02 \x02(\x0b\x32\x0b.protos.DNN'
+    ),
+    dependencies=[
+        easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
+    ])
+
+_UNITERTOWER = _descriptor.Descriptor(
+    name='UniterTower',
+    full_name='protos.UniterTower',
+    filename=None,
+    file=DESCRIPTOR,
+    containing_type=None,
+    fields=[
+        _descriptor.FieldDescriptor(
+            name='hidden_size',
+            full_name='protos.UniterTower.hidden_size',
+            index=0,
+            number=1,
+            type=13,
+            cpp_type=3,
+            label=2,
+            has_default_value=False,
+            default_value=0,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='num_hidden_layers',
+            full_name='protos.UniterTower.num_hidden_layers',
+            index=1,
+            number=2,
+            type=13,
+            cpp_type=3,
+            label=2,
+            has_default_value=False,
+            default_value=0,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='num_attention_heads',
+            full_name='protos.UniterTower.num_attention_heads',
+            index=2,
+            number=3,
+            type=13,
+            cpp_type=3,
+            label=2,
+            has_default_value=False,
+            default_value=0,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='intermediate_size',
+            full_name='protos.UniterTower.intermediate_size',
+            index=3,
+            number=4,
+            type=13,
+            cpp_type=3,
+            label=2,
+            has_default_value=False,
+            default_value=0,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='hidden_act',
+            full_name='protos.UniterTower.hidden_act',
+            index=4,
+            number=5,
+            type=9,
+            cpp_type=9,
+            label=2,
+            has_default_value=True,
+            default_value=_b('gelu').decode('utf-8'),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='hidden_dropout_prob',
+            full_name='protos.UniterTower.hidden_dropout_prob',
+            index=5,
+            number=6,
+            type=2,
+            cpp_type=6,
+            label=2,
+            has_default_value=True,
+            default_value=float(0.1),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='attention_probs_dropout_prob',
+            full_name='protos.UniterTower.attention_probs_dropout_prob',
+            index=6,
+            number=7,
+            type=2,
+            cpp_type=6,
+            label=2,
+            has_default_value=True,
+            default_value=float(0.1),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='max_position_embeddings',
+            full_name='protos.UniterTower.max_position_embeddings',
+            index=7,
+            number=8,
+            type=13,
+            cpp_type=3,
+            label=2,
+            has_default_value=True,
+            default_value=512,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='use_position_embeddings',
+            full_name='protos.UniterTower.use_position_embeddings',
+            index=8,
+            number=9,
+            type=8,
+            cpp_type=7,
+            label=2,
+            has_default_value=True,
+            default_value=True,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='initializer_range',
+            full_name='protos.UniterTower.initializer_range',
+            index=9,
+            number=10,
+            type=2,
+            cpp_type=6,
+            label=2,
+            has_default_value=True,
+            default_value=float(0.02),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='other_feature_dnn',
+            full_name='protos.UniterTower.other_feature_dnn',
+            index=10,
+            number=11,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+    ],
+    extensions=[],
+    nested_types=[],
+    enum_types=[],
+    options=None,
+    is_extendable=False,
+    syntax='proto2',
+    extension_ranges=[],
+    oneofs=[],
+    serialized_start=82,
+    serialized_end=452,
+)
+
+_UNITER = _descriptor.Descriptor(
+    name='Uniter',
+    full_name='protos.Uniter',
+    filename=None,
+    file=DESCRIPTOR,
+    containing_type=None,
+    fields=[
+        _descriptor.FieldDescriptor(
+            name='config',
+            full_name='protos.Uniter.config',
+            index=0,
+            number=1,
+            type=11,
+            cpp_type=10,
+            label=2,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='final_dnn',
+            full_name='protos.Uniter.final_dnn',
+            index=1,
+            number=2,
+            type=11,
+            cpp_type=10,
+            label=2,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+    ],
+    extensions=[],
+    nested_types=[],
+    enum_types=[],
+    options=None,
+    is_extendable=False,
+    syntax='proto2',
+    extension_ranges=[],
+    oneofs=[],
+    serialized_start=454,
+    serialized_end=531,
+)
+
+_UNITERTOWER.fields_by_name[
+    'other_feature_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
+_UNITER.fields_by_name['config'].message_type = _UNITERTOWER
+_UNITER.fields_by_name[
+    'final_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
+DESCRIPTOR.message_types_by_name['UniterTower'] = _UNITERTOWER
+DESCRIPTOR.message_types_by_name['Uniter'] = _UNITER
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+UniterTower = _reflection.GeneratedProtocolMessageType(
+    'UniterTower',
+    (_message.Message,),
+    dict(
+        DESCRIPTOR=_UNITERTOWER,
+        __module__='easy_rec.python.protos.uniter_pb2'
+        # @@protoc_insertion_point(class_scope:protos.UniterTower)
+    ))
+_sym_db.RegisterMessage(UniterTower)
+
+Uniter = _reflection.GeneratedProtocolMessageType(
+    'Uniter',
+    (_message.Message,),
+    dict(
+        DESCRIPTOR=_UNITER,
+        __module__='easy_rec.python.protos.uniter_pb2'
+        # @@protoc_insertion_point(class_scope:protos.Uniter)
+    ))
+_sym_db.RegisterMessage(Uniter)
+
+# @@protoc_insertion_point(module_scope)

easy_rec/python/protos/variational_dropout_pb2.py

@@ -0,0 +1,91 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: easy_rec/python/protos/variational_dropout.proto
+
+import sys
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pb2
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+
+_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+    name='easy_rec/python/protos/variational_dropout.proto',
+    package='protos',
+    syntax='proto2',
+    serialized_pb=_b(
+        '\n0easy_rec/python/protos/variational_dropout.proto\x12\x06protos\"q\n\x17VariationalDropoutLayer\x12#\n\x15regularization_lambda\x18\x01 \x01(\x02:\x04\x30.01\x12\x31\n\"embedding_wise_variational_dropout\x18\x02 \x01(\x08:\x05\x66\x61lse'
+    ))
+
+_VARIATIONALDROPOUTLAYER = _descriptor.Descriptor(
+    name='VariationalDropoutLayer',
+    full_name='protos.VariationalDropoutLayer',
+    filename=None,
+    file=DESCRIPTOR,
+    containing_type=None,
+    fields=[
+        _descriptor.FieldDescriptor(
+            name='regularization_lambda',
+            full_name='protos.VariationalDropoutLayer.regularization_lambda',
+            index=0,
+            number=1,
+            type=2,
+            cpp_type=6,
+            label=1,
+            has_default_value=True,
+            default_value=float(0.01),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='embedding_wise_variational_dropout',
+            full_name='protos.VariationalDropoutLayer.embedding_wise_variational_dropout',
+            index=1,
+            number=2,
+            type=8,
+            cpp_type=7,
+            label=1,
+            has_default_value=True,
+            default_value=False,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+    ],
+    extensions=[],
+    nested_types=[],
+    enum_types=[],
+    options=None,
+    is_extendable=False,
+    syntax='proto2',
+    extension_ranges=[],
+    oneofs=[],
+    serialized_start=60,
+    serialized_end=173,
+)
+
+DESCRIPTOR.message_types_by_name[
+    'VariationalDropoutLayer'] = _VARIATIONALDROPOUTLAYER
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+VariationalDropoutLayer = _reflection.GeneratedProtocolMessageType(
+    'VariationalDropoutLayer',
+    (_message.Message,),
+    dict(
+        DESCRIPTOR=_VARIATIONALDROPOUTLAYER,
+        __module__='easy_rec.python.protos.variational_dropout_pb2'
+        # @@protoc_insertion_point(class_scope:protos.VariationalDropoutLayer)
+    ))
+_sym_db.RegisterMessage(VariationalDropoutLayer)
+
+# @@protoc_insertion_point(module_scope)

easy_rec/python/protos/wide_and_deep_pb2.py

@@ -0,0 +1,131 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: easy_rec/python/protos/wide_and_deep.proto
+
+import sys
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pb2
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+
+from easy_rec.python.protos import dnn_pb2 as easy__rec_dot_python_dot_protos_dot_dnn__pb2  # NOQA
+
+_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode('latin1'))
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+    name='easy_rec/python/protos/wide_and_deep.proto',
+    package='protos',
+    syntax='proto2',
+    serialized_pb=_b(
+        '\n*easy_rec/python/protos/wide_and_deep.proto\x12\x06protos\x1a easy_rec/python/protos/dnn.proto\"\x86\x01\n\x0bWideAndDeep\x12\x1a\n\x0fwide_output_dim\x18\x01 \x02(\r:\x01\x31\x12\x18\n\x03\x64nn\x18\x02 \x02(\x0b\x32\x0b.protos.DNN\x12\x1e\n\tfinal_dnn\x18\x03 \x01(\x0b\x32\x0b.protos.DNN\x12!\n\x11l2_regularization\x18\x04 \x01(\x02:\x06\x30.0001'
+    ),
+    dependencies=[
+        easy__rec_dot_python_dot_protos_dot_dnn__pb2.DESCRIPTOR,
+    ])
+
+_WIDEANDDEEP = _descriptor.Descriptor(
+    name='WideAndDeep',
+    full_name='protos.WideAndDeep',
+    filename=None,
+    file=DESCRIPTOR,
+    containing_type=None,
+    fields=[
+        _descriptor.FieldDescriptor(
+            name='wide_output_dim',
+            full_name='protos.WideAndDeep.wide_output_dim',
+            index=0,
+            number=1,
+            type=13,
+            cpp_type=3,
+            label=2,
+            has_default_value=True,
+            default_value=1,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='dnn',
+            full_name='protos.WideAndDeep.dnn',
+            index=1,
+            number=2,
+            type=11,
+            cpp_type=10,
+            label=2,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='final_dnn',
+            full_name='protos.WideAndDeep.final_dnn',
+            index=2,
+            number=3,
+            type=11,
+            cpp_type=10,
+            label=1,
+            has_default_value=False,
+            default_value=None,
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+        _descriptor.FieldDescriptor(
+            name='l2_regularization',
+            full_name='protos.WideAndDeep.l2_regularization',
+            index=3,
+            number=4,
+            type=2,
+            cpp_type=6,
+            label=1,
+            has_default_value=True,
+            default_value=float(0.0001),
+            message_type=None,
+            enum_type=None,
+            containing_type=None,
+            is_extension=False,
+            extension_scope=None,
+            options=None),
+    ],
+    extensions=[],
+    nested_types=[],
+    enum_types=[],
+    options=None,
+    is_extendable=False,
+    syntax='proto2',
+    extension_ranges=[],
+    oneofs=[],
+    serialized_start=89,
+    serialized_end=223,
+)
+
+_WIDEANDDEEP.fields_by_name[
+    'dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
+_WIDEANDDEEP.fields_by_name[
+    'final_dnn'].message_type = easy__rec_dot_python_dot_protos_dot_dnn__pb2._DNN
+DESCRIPTOR.message_types_by_name['WideAndDeep'] = _WIDEANDDEEP
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+WideAndDeep = _reflection.GeneratedProtocolMessageType(
+    'WideAndDeep',
+    (_message.Message,),
+    dict(
+        DESCRIPTOR=_WIDEANDDEEP,
+        __module__='easy_rec.python.protos.wide_and_deep_pb2'
+        # @@protoc_insertion_point(class_scope:protos.WideAndDeep)
+    ))
+_sym_db.RegisterMessage(WideAndDeep)
+
+# @@protoc_insertion_point(module_scope)