├── .gitignore ├── ArrayFeatureExtractor_pb2.py ├── CategoricalMapping_pb2.py ├── DataStructures_pb2.py ├── DictVectorizer_pb2.py ├── FeatureTypes_pb2.py ├── FeatureVectorizer_pb2.py ├── GLMClassifier_pb2.py ├── GLMRegressor_pb2.py ├── Identity_pb2.py ├── Imputer_pb2.py ├── Model_pb2.py ├── NeuralNetwork_pb2.py ├── Normalizer_pb2.py ├── OneHotEncoder_pb2.py ├── README.markdown ├── SVM_pb2.py ├── Scaler_pb2.py ├── TreeEnsemble_pb2.py └── lookieloo.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | 3 | # Byte-compiled / optimized / DLL files 4 | __pycache__/ 5 | *.py[cod] 6 | *$py.class 7 | 8 | # C extensions 9 | *.so 10 | 11 | # Distribution / packaging 12 | .Python 13 | env/ 14 | build/ 15 | develop-eggs/ 16 | dist/ 17 | downloads/ 18 | eggs/ 19 | .eggs/ 20 | lib/ 21 | lib64/ 22 | parts/ 23 | sdist/ 24 | var/ 25 | wheels/ 26 | *.egg-info/ 27 | .installed.cfg 28 | *.egg 29 | 30 | # PyInstaller 31 | # Usually these files are written by a python script from a template 32 | # before PyInstaller builds the exe, so as to inject date/other infos into it. 33 | *.manifest 34 | *.spec 35 | 36 | # Installer logs 37 | pip-log.txt 38 | pip-delete-this-directory.txt 39 | 40 | # Unit test / coverage reports 41 | htmlcov/ 42 | .tox/ 43 | .coverage 44 | .coverage.* 45 | .cache 46 | nosetests.xml 47 | coverage.xml 48 | *,cover 49 | .hypothesis/ 50 | 51 | # Translations 52 | *.mo 53 | *.pot 54 | 55 | # Django stuff: 56 | *.log 57 | local_settings.py 58 | 59 | # Flask stuff: 60 | instance/ 61 | .webassets-cache 62 | 63 | # Scrapy stuff: 64 | .scrapy 65 | 66 | # Sphinx documentation 67 | docs/_build/ 68 | 69 | # PyBuilder 70 | target/ 71 | 72 | # Jupyter Notebook 73 | .ipynb_checkpoints 74 | 75 | # pyenv 76 | .python-version 77 | 78 | # celery beat schedule file 79 | celerybeat-schedule 80 | 81 | # SageMath parsed files 82 | *.sage.py 83 | 84 | # dotenv 85 | .env 86 | 87 | # virtualenv 88 | .venv 89 | venv/ 90 | ENV/ 91 | 92 | # Spyder project settings 93 | .spyderproject 94 | .spyproject 95 | 96 | # Rope project settings 97 | .ropeproject 98 | 99 | # mkdocs documentation 100 | /site 101 | -------------------------------------------------------------------------------- /ArrayFeatureExtractor_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: ArrayFeatureExtractor.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='ArrayFeatureExtractor.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x1b\x41rrayFeatureExtractor.proto\x12\x14\x43oreML.Specification\"-\n\x15\x41rrayFeatureExtractor\x12\x14\n\x0c\x65xtractIndex\x18\x01 \x03(\x04\x42\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | 29 | _ARRAYFEATUREEXTRACTOR = _descriptor.Descriptor( 30 | name='ArrayFeatureExtractor', 31 | full_name='CoreML.Specification.ArrayFeatureExtractor', 32 | filename=None, 33 | file=DESCRIPTOR, 34 | containing_type=None, 35 | fields=[ 36 | _descriptor.FieldDescriptor( 37 | name='extractIndex', full_name='CoreML.Specification.ArrayFeatureExtractor.extractIndex', index=0, 38 | number=1, type=4, cpp_type=4, label=3, 39 | has_default_value=False, default_value=[], 40 | message_type=None, enum_type=None, containing_type=None, 41 | is_extension=False, extension_scope=None, 42 | options=None), 43 | ], 44 | extensions=[ 45 | ], 46 | nested_types=[], 47 | enum_types=[ 48 | ], 49 | options=None, 50 | is_extendable=False, 51 | syntax='proto3', 52 | extension_ranges=[], 53 | oneofs=[ 54 | ], 55 | serialized_start=53, 56 | serialized_end=98, 57 | ) 58 | 59 | DESCRIPTOR.message_types_by_name['ArrayFeatureExtractor'] = _ARRAYFEATUREEXTRACTOR 60 | 61 | ArrayFeatureExtractor = _reflection.GeneratedProtocolMessageType('ArrayFeatureExtractor', (_message.Message,), dict( 62 | DESCRIPTOR = _ARRAYFEATUREEXTRACTOR, 63 | __module__ = 'ArrayFeatureExtractor_pb2' 64 | # @@protoc_insertion_point(class_scope:CoreML.Specification.ArrayFeatureExtractor) 65 | )) 66 | _sym_db.RegisterMessage(ArrayFeatureExtractor) 67 | 68 | 69 | DESCRIPTOR.has_options = True 70 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 71 | # @@protoc_insertion_point(module_scope) 72 | -------------------------------------------------------------------------------- /CategoricalMapping_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: CategoricalMapping.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import DataStructures_pb2 as DataStructures__pb2 17 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 18 | 19 | from DataStructures_pb2 import * 20 | 21 | DESCRIPTOR = _descriptor.FileDescriptor( 22 | name='CategoricalMapping.proto', 23 | package='CoreML.Specification', 24 | syntax='proto3', 25 | serialized_pb=_b('\n\x18\x43\x61tegoricalMapping.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\xe7\x01\n\x12\x43\x61tegoricalMapping\x12\x42\n\x10stringToInt64Map\x18\x01 \x01(\x0b\x32&.CoreML.Specification.StringToInt64MapH\x00\x12\x42\n\x10int64ToStringMap\x18\x02 \x01(\x0b\x32&.CoreML.Specification.Int64ToStringMapH\x00\x12\x12\n\x08strValue\x18\x65 \x01(\tH\x01\x12\x14\n\nint64Value\x18\x66 \x01(\x03H\x01\x42\r\n\x0bMappingTypeB\x10\n\x0eValueOnUnknownB\x02H\x03P\x00\x62\x06proto3') 26 | , 27 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 28 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 29 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 30 | 31 | 32 | 33 | 34 | _CATEGORICALMAPPING = _descriptor.Descriptor( 35 | name='CategoricalMapping', 36 | full_name='CoreML.Specification.CategoricalMapping', 37 | filename=None, 38 | file=DESCRIPTOR, 39 | containing_type=None, 40 | fields=[ 41 | _descriptor.FieldDescriptor( 42 | name='stringToInt64Map', full_name='CoreML.Specification.CategoricalMapping.stringToInt64Map', index=0, 43 | number=1, type=11, cpp_type=10, label=1, 44 | has_default_value=False, default_value=None, 45 | message_type=None, enum_type=None, containing_type=None, 46 | is_extension=False, extension_scope=None, 47 | options=None), 48 | _descriptor.FieldDescriptor( 49 | name='int64ToStringMap', full_name='CoreML.Specification.CategoricalMapping.int64ToStringMap', index=1, 50 | number=2, type=11, cpp_type=10, label=1, 51 | has_default_value=False, default_value=None, 52 | message_type=None, enum_type=None, containing_type=None, 53 | is_extension=False, extension_scope=None, 54 | options=None), 55 | _descriptor.FieldDescriptor( 56 | name='strValue', full_name='CoreML.Specification.CategoricalMapping.strValue', index=2, 57 | number=101, type=9, cpp_type=9, label=1, 58 | has_default_value=False, default_value=_b("").decode('utf-8'), 59 | message_type=None, enum_type=None, containing_type=None, 60 | is_extension=False, extension_scope=None, 61 | options=None), 62 | _descriptor.FieldDescriptor( 63 | name='int64Value', full_name='CoreML.Specification.CategoricalMapping.int64Value', index=3, 64 | number=102, type=3, cpp_type=2, label=1, 65 | has_default_value=False, default_value=0, 66 | message_type=None, enum_type=None, containing_type=None, 67 | is_extension=False, extension_scope=None, 68 | options=None), 69 | ], 70 | extensions=[ 71 | ], 72 | nested_types=[], 73 | enum_types=[ 74 | ], 75 | options=None, 76 | is_extendable=False, 77 | syntax='proto3', 78 | extension_ranges=[], 79 | oneofs=[ 80 | _descriptor.OneofDescriptor( 81 | name='MappingType', full_name='CoreML.Specification.CategoricalMapping.MappingType', 82 | index=0, containing_type=None, fields=[]), 83 | _descriptor.OneofDescriptor( 84 | name='ValueOnUnknown', full_name='CoreML.Specification.CategoricalMapping.ValueOnUnknown', 85 | index=1, containing_type=None, fields=[]), 86 | ], 87 | serialized_start=73, 88 | serialized_end=304, 89 | ) 90 | 91 | _CATEGORICALMAPPING.fields_by_name['stringToInt64Map'].message_type = DataStructures__pb2._STRINGTOINT64MAP 92 | _CATEGORICALMAPPING.fields_by_name['int64ToStringMap'].message_type = DataStructures__pb2._INT64TOSTRINGMAP 93 | _CATEGORICALMAPPING.oneofs_by_name['MappingType'].fields.append( 94 | _CATEGORICALMAPPING.fields_by_name['stringToInt64Map']) 95 | _CATEGORICALMAPPING.fields_by_name['stringToInt64Map'].containing_oneof = _CATEGORICALMAPPING.oneofs_by_name['MappingType'] 96 | _CATEGORICALMAPPING.oneofs_by_name['MappingType'].fields.append( 97 | _CATEGORICALMAPPING.fields_by_name['int64ToStringMap']) 98 | _CATEGORICALMAPPING.fields_by_name['int64ToStringMap'].containing_oneof = _CATEGORICALMAPPING.oneofs_by_name['MappingType'] 99 | _CATEGORICALMAPPING.oneofs_by_name['ValueOnUnknown'].fields.append( 100 | _CATEGORICALMAPPING.fields_by_name['strValue']) 101 | _CATEGORICALMAPPING.fields_by_name['strValue'].containing_oneof = _CATEGORICALMAPPING.oneofs_by_name['ValueOnUnknown'] 102 | _CATEGORICALMAPPING.oneofs_by_name['ValueOnUnknown'].fields.append( 103 | _CATEGORICALMAPPING.fields_by_name['int64Value']) 104 | _CATEGORICALMAPPING.fields_by_name['int64Value'].containing_oneof = _CATEGORICALMAPPING.oneofs_by_name['ValueOnUnknown'] 105 | DESCRIPTOR.message_types_by_name['CategoricalMapping'] = _CATEGORICALMAPPING 106 | 107 | CategoricalMapping = _reflection.GeneratedProtocolMessageType('CategoricalMapping', (_message.Message,), dict( 108 | DESCRIPTOR = _CATEGORICALMAPPING, 109 | __module__ = 'CategoricalMapping_pb2' 110 | # @@protoc_insertion_point(class_scope:CoreML.Specification.CategoricalMapping) 111 | )) 112 | _sym_db.RegisterMessage(CategoricalMapping) 113 | 114 | 115 | DESCRIPTOR.has_options = True 116 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 117 | # @@protoc_insertion_point(module_scope) 118 | -------------------------------------------------------------------------------- /DataStructures_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: DataStructures.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import FeatureTypes_pb2 as FeatureTypes__pb2 17 | 18 | from FeatureTypes_pb2 import * 19 | 20 | DESCRIPTOR = _descriptor.FileDescriptor( 21 | name='DataStructures.proto', 22 | package='CoreML.Specification', 23 | syntax='proto3', 24 | serialized_pb=_b('\n\x14\x44\x61taStructures.proto\x12\x14\x43oreML.Specification\x1a\x12\x46\x65\x61tureTypes.proto\"|\n\x10StringToInt64Map\x12<\n\x03map\x18\x01 \x03(\x0b\x32/.CoreML.Specification.StringToInt64Map.MapEntry\x1a*\n\x08MapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x03:\x02\x38\x01\"|\n\x10Int64ToStringMap\x12<\n\x03map\x18\x01 \x03(\x0b\x32/.CoreML.Specification.Int64ToStringMap.MapEntry\x1a*\n\x08MapEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"~\n\x11StringToDoubleMap\x12=\n\x03map\x18\x01 \x03(\x0b\x32\x30.CoreML.Specification.StringToDoubleMap.MapEntry\x1a*\n\x08MapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"|\n\x10Int64ToDoubleMap\x12<\n\x03map\x18\x01 \x03(\x0b\x32/.CoreML.Specification.Int64ToDoubleMap.MapEntry\x1a*\n\x08MapEntry\x12\x0b\n\x03key\x18\x01 \x01(\x03\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\x1e\n\x0cStringVector\x12\x0e\n\x06vector\x18\x01 \x03(\t\"\x1d\n\x0bInt64Vector\x12\x0e\n\x06vector\x18\x01 \x03(\x03\"\x1e\n\x0c\x44oubleVector\x12\x0e\n\x06vector\x18\x01 \x03(\x01\x42\x02H\x03P\x00\x62\x06proto3') 25 | , 26 | dependencies=[FeatureTypes__pb2.DESCRIPTOR,], 27 | public_dependencies=[FeatureTypes__pb2.DESCRIPTOR,]) 28 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 29 | 30 | 31 | 32 | 33 | _STRINGTOINT64MAP_MAPENTRY = _descriptor.Descriptor( 34 | name='MapEntry', 35 | full_name='CoreML.Specification.StringToInt64Map.MapEntry', 36 | filename=None, 37 | file=DESCRIPTOR, 38 | containing_type=None, 39 | fields=[ 40 | _descriptor.FieldDescriptor( 41 | name='key', full_name='CoreML.Specification.StringToInt64Map.MapEntry.key', index=0, 42 | number=1, type=9, cpp_type=9, label=1, 43 | has_default_value=False, default_value=_b("").decode('utf-8'), 44 | message_type=None, enum_type=None, containing_type=None, 45 | is_extension=False, extension_scope=None, 46 | options=None), 47 | _descriptor.FieldDescriptor( 48 | name='value', full_name='CoreML.Specification.StringToInt64Map.MapEntry.value', index=1, 49 | number=2, type=3, cpp_type=2, label=1, 50 | has_default_value=False, default_value=0, 51 | message_type=None, enum_type=None, containing_type=None, 52 | is_extension=False, extension_scope=None, 53 | options=None), 54 | ], 55 | extensions=[ 56 | ], 57 | nested_types=[], 58 | enum_types=[ 59 | ], 60 | options=_descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')), 61 | is_extendable=False, 62 | syntax='proto3', 63 | extension_ranges=[], 64 | oneofs=[ 65 | ], 66 | serialized_start=148, 67 | serialized_end=190, 68 | ) 69 | 70 | _STRINGTOINT64MAP = _descriptor.Descriptor( 71 | name='StringToInt64Map', 72 | full_name='CoreML.Specification.StringToInt64Map', 73 | filename=None, 74 | file=DESCRIPTOR, 75 | containing_type=None, 76 | fields=[ 77 | _descriptor.FieldDescriptor( 78 | name='map', full_name='CoreML.Specification.StringToInt64Map.map', index=0, 79 | number=1, type=11, cpp_type=10, label=3, 80 | has_default_value=False, default_value=[], 81 | message_type=None, enum_type=None, containing_type=None, 82 | is_extension=False, extension_scope=None, 83 | options=None), 84 | ], 85 | extensions=[ 86 | ], 87 | nested_types=[_STRINGTOINT64MAP_MAPENTRY, ], 88 | enum_types=[ 89 | ], 90 | options=None, 91 | is_extendable=False, 92 | syntax='proto3', 93 | extension_ranges=[], 94 | oneofs=[ 95 | ], 96 | serialized_start=66, 97 | serialized_end=190, 98 | ) 99 | 100 | 101 | _INT64TOSTRINGMAP_MAPENTRY = _descriptor.Descriptor( 102 | name='MapEntry', 103 | full_name='CoreML.Specification.Int64ToStringMap.MapEntry', 104 | filename=None, 105 | file=DESCRIPTOR, 106 | containing_type=None, 107 | fields=[ 108 | _descriptor.FieldDescriptor( 109 | name='key', full_name='CoreML.Specification.Int64ToStringMap.MapEntry.key', index=0, 110 | number=1, type=3, cpp_type=2, label=1, 111 | has_default_value=False, default_value=0, 112 | message_type=None, enum_type=None, containing_type=None, 113 | is_extension=False, extension_scope=None, 114 | options=None), 115 | _descriptor.FieldDescriptor( 116 | name='value', full_name='CoreML.Specification.Int64ToStringMap.MapEntry.value', index=1, 117 | number=2, type=9, cpp_type=9, label=1, 118 | has_default_value=False, default_value=_b("").decode('utf-8'), 119 | message_type=None, enum_type=None, containing_type=None, 120 | is_extension=False, extension_scope=None, 121 | options=None), 122 | ], 123 | extensions=[ 124 | ], 125 | nested_types=[], 126 | enum_types=[ 127 | ], 128 | options=_descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')), 129 | is_extendable=False, 130 | syntax='proto3', 131 | extension_ranges=[], 132 | oneofs=[ 133 | ], 134 | serialized_start=274, 135 | serialized_end=316, 136 | ) 137 | 138 | _INT64TOSTRINGMAP = _descriptor.Descriptor( 139 | name='Int64ToStringMap', 140 | full_name='CoreML.Specification.Int64ToStringMap', 141 | filename=None, 142 | file=DESCRIPTOR, 143 | containing_type=None, 144 | fields=[ 145 | _descriptor.FieldDescriptor( 146 | name='map', full_name='CoreML.Specification.Int64ToStringMap.map', index=0, 147 | number=1, type=11, cpp_type=10, label=3, 148 | has_default_value=False, default_value=[], 149 | message_type=None, enum_type=None, containing_type=None, 150 | is_extension=False, extension_scope=None, 151 | options=None), 152 | ], 153 | extensions=[ 154 | ], 155 | nested_types=[_INT64TOSTRINGMAP_MAPENTRY, ], 156 | enum_types=[ 157 | ], 158 | options=None, 159 | is_extendable=False, 160 | syntax='proto3', 161 | extension_ranges=[], 162 | oneofs=[ 163 | ], 164 | serialized_start=192, 165 | serialized_end=316, 166 | ) 167 | 168 | 169 | _STRINGTODOUBLEMAP_MAPENTRY = _descriptor.Descriptor( 170 | name='MapEntry', 171 | full_name='CoreML.Specification.StringToDoubleMap.MapEntry', 172 | filename=None, 173 | file=DESCRIPTOR, 174 | containing_type=None, 175 | fields=[ 176 | _descriptor.FieldDescriptor( 177 | name='key', full_name='CoreML.Specification.StringToDoubleMap.MapEntry.key', index=0, 178 | number=1, type=9, cpp_type=9, label=1, 179 | has_default_value=False, default_value=_b("").decode('utf-8'), 180 | message_type=None, enum_type=None, containing_type=None, 181 | is_extension=False, extension_scope=None, 182 | options=None), 183 | _descriptor.FieldDescriptor( 184 | name='value', full_name='CoreML.Specification.StringToDoubleMap.MapEntry.value', index=1, 185 | number=2, type=1, cpp_type=5, label=1, 186 | has_default_value=False, default_value=float(0), 187 | message_type=None, enum_type=None, containing_type=None, 188 | is_extension=False, extension_scope=None, 189 | options=None), 190 | ], 191 | extensions=[ 192 | ], 193 | nested_types=[], 194 | enum_types=[ 195 | ], 196 | options=_descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')), 197 | is_extendable=False, 198 | syntax='proto3', 199 | extension_ranges=[], 200 | oneofs=[ 201 | ], 202 | serialized_start=402, 203 | serialized_end=444, 204 | ) 205 | 206 | _STRINGTODOUBLEMAP = _descriptor.Descriptor( 207 | name='StringToDoubleMap', 208 | full_name='CoreML.Specification.StringToDoubleMap', 209 | filename=None, 210 | file=DESCRIPTOR, 211 | containing_type=None, 212 | fields=[ 213 | _descriptor.FieldDescriptor( 214 | name='map', full_name='CoreML.Specification.StringToDoubleMap.map', index=0, 215 | number=1, type=11, cpp_type=10, label=3, 216 | has_default_value=False, default_value=[], 217 | message_type=None, enum_type=None, containing_type=None, 218 | is_extension=False, extension_scope=None, 219 | options=None), 220 | ], 221 | extensions=[ 222 | ], 223 | nested_types=[_STRINGTODOUBLEMAP_MAPENTRY, ], 224 | enum_types=[ 225 | ], 226 | options=None, 227 | is_extendable=False, 228 | syntax='proto3', 229 | extension_ranges=[], 230 | oneofs=[ 231 | ], 232 | serialized_start=318, 233 | serialized_end=444, 234 | ) 235 | 236 | 237 | _INT64TODOUBLEMAP_MAPENTRY = _descriptor.Descriptor( 238 | name='MapEntry', 239 | full_name='CoreML.Specification.Int64ToDoubleMap.MapEntry', 240 | filename=None, 241 | file=DESCRIPTOR, 242 | containing_type=None, 243 | fields=[ 244 | _descriptor.FieldDescriptor( 245 | name='key', full_name='CoreML.Specification.Int64ToDoubleMap.MapEntry.key', index=0, 246 | number=1, type=3, cpp_type=2, label=1, 247 | has_default_value=False, default_value=0, 248 | message_type=None, enum_type=None, containing_type=None, 249 | is_extension=False, extension_scope=None, 250 | options=None), 251 | _descriptor.FieldDescriptor( 252 | name='value', full_name='CoreML.Specification.Int64ToDoubleMap.MapEntry.value', index=1, 253 | number=2, type=1, cpp_type=5, label=1, 254 | has_default_value=False, default_value=float(0), 255 | message_type=None, enum_type=None, containing_type=None, 256 | is_extension=False, extension_scope=None, 257 | options=None), 258 | ], 259 | extensions=[ 260 | ], 261 | nested_types=[], 262 | enum_types=[ 263 | ], 264 | options=_descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')), 265 | is_extendable=False, 266 | syntax='proto3', 267 | extension_ranges=[], 268 | oneofs=[ 269 | ], 270 | serialized_start=528, 271 | serialized_end=570, 272 | ) 273 | 274 | _INT64TODOUBLEMAP = _descriptor.Descriptor( 275 | name='Int64ToDoubleMap', 276 | full_name='CoreML.Specification.Int64ToDoubleMap', 277 | filename=None, 278 | file=DESCRIPTOR, 279 | containing_type=None, 280 | fields=[ 281 | _descriptor.FieldDescriptor( 282 | name='map', full_name='CoreML.Specification.Int64ToDoubleMap.map', index=0, 283 | number=1, type=11, cpp_type=10, label=3, 284 | has_default_value=False, default_value=[], 285 | message_type=None, enum_type=None, containing_type=None, 286 | is_extension=False, extension_scope=None, 287 | options=None), 288 | ], 289 | extensions=[ 290 | ], 291 | nested_types=[_INT64TODOUBLEMAP_MAPENTRY, ], 292 | enum_types=[ 293 | ], 294 | options=None, 295 | is_extendable=False, 296 | syntax='proto3', 297 | extension_ranges=[], 298 | oneofs=[ 299 | ], 300 | serialized_start=446, 301 | serialized_end=570, 302 | ) 303 | 304 | 305 | _STRINGVECTOR = _descriptor.Descriptor( 306 | name='StringVector', 307 | full_name='CoreML.Specification.StringVector', 308 | filename=None, 309 | file=DESCRIPTOR, 310 | containing_type=None, 311 | fields=[ 312 | _descriptor.FieldDescriptor( 313 | name='vector', full_name='CoreML.Specification.StringVector.vector', index=0, 314 | number=1, type=9, cpp_type=9, label=3, 315 | has_default_value=False, default_value=[], 316 | message_type=None, enum_type=None, containing_type=None, 317 | is_extension=False, extension_scope=None, 318 | options=None), 319 | ], 320 | extensions=[ 321 | ], 322 | nested_types=[], 323 | enum_types=[ 324 | ], 325 | options=None, 326 | is_extendable=False, 327 | syntax='proto3', 328 | extension_ranges=[], 329 | oneofs=[ 330 | ], 331 | serialized_start=572, 332 | serialized_end=602, 333 | ) 334 | 335 | 336 | _INT64VECTOR = _descriptor.Descriptor( 337 | name='Int64Vector', 338 | full_name='CoreML.Specification.Int64Vector', 339 | filename=None, 340 | file=DESCRIPTOR, 341 | containing_type=None, 342 | fields=[ 343 | _descriptor.FieldDescriptor( 344 | name='vector', full_name='CoreML.Specification.Int64Vector.vector', index=0, 345 | number=1, type=3, cpp_type=2, label=3, 346 | has_default_value=False, default_value=[], 347 | message_type=None, enum_type=None, containing_type=None, 348 | is_extension=False, extension_scope=None, 349 | options=None), 350 | ], 351 | extensions=[ 352 | ], 353 | nested_types=[], 354 | enum_types=[ 355 | ], 356 | options=None, 357 | is_extendable=False, 358 | syntax='proto3', 359 | extension_ranges=[], 360 | oneofs=[ 361 | ], 362 | serialized_start=604, 363 | serialized_end=633, 364 | ) 365 | 366 | 367 | _DOUBLEVECTOR = _descriptor.Descriptor( 368 | name='DoubleVector', 369 | full_name='CoreML.Specification.DoubleVector', 370 | filename=None, 371 | file=DESCRIPTOR, 372 | containing_type=None, 373 | fields=[ 374 | _descriptor.FieldDescriptor( 375 | name='vector', full_name='CoreML.Specification.DoubleVector.vector', index=0, 376 | number=1, type=1, cpp_type=5, label=3, 377 | has_default_value=False, default_value=[], 378 | message_type=None, enum_type=None, containing_type=None, 379 | is_extension=False, extension_scope=None, 380 | options=None), 381 | ], 382 | extensions=[ 383 | ], 384 | nested_types=[], 385 | enum_types=[ 386 | ], 387 | options=None, 388 | is_extendable=False, 389 | syntax='proto3', 390 | extension_ranges=[], 391 | oneofs=[ 392 | ], 393 | serialized_start=635, 394 | serialized_end=665, 395 | ) 396 | 397 | _STRINGTOINT64MAP_MAPENTRY.containing_type = _STRINGTOINT64MAP 398 | _STRINGTOINT64MAP.fields_by_name['map'].message_type = _STRINGTOINT64MAP_MAPENTRY 399 | _INT64TOSTRINGMAP_MAPENTRY.containing_type = _INT64TOSTRINGMAP 400 | _INT64TOSTRINGMAP.fields_by_name['map'].message_type = _INT64TOSTRINGMAP_MAPENTRY 401 | _STRINGTODOUBLEMAP_MAPENTRY.containing_type = _STRINGTODOUBLEMAP 402 | _STRINGTODOUBLEMAP.fields_by_name['map'].message_type = _STRINGTODOUBLEMAP_MAPENTRY 403 | _INT64TODOUBLEMAP_MAPENTRY.containing_type = _INT64TODOUBLEMAP 404 | _INT64TODOUBLEMAP.fields_by_name['map'].message_type = _INT64TODOUBLEMAP_MAPENTRY 405 | DESCRIPTOR.message_types_by_name['StringToInt64Map'] = _STRINGTOINT64MAP 406 | DESCRIPTOR.message_types_by_name['Int64ToStringMap'] = _INT64TOSTRINGMAP 407 | DESCRIPTOR.message_types_by_name['StringToDoubleMap'] = _STRINGTODOUBLEMAP 408 | DESCRIPTOR.message_types_by_name['Int64ToDoubleMap'] = _INT64TODOUBLEMAP 409 | DESCRIPTOR.message_types_by_name['StringVector'] = _STRINGVECTOR 410 | DESCRIPTOR.message_types_by_name['Int64Vector'] = _INT64VECTOR 411 | DESCRIPTOR.message_types_by_name['DoubleVector'] = _DOUBLEVECTOR 412 | 413 | StringToInt64Map = _reflection.GeneratedProtocolMessageType('StringToInt64Map', (_message.Message,), dict( 414 | 415 | MapEntry = _reflection.GeneratedProtocolMessageType('MapEntry', (_message.Message,), dict( 416 | DESCRIPTOR = _STRINGTOINT64MAP_MAPENTRY, 417 | __module__ = 'DataStructures_pb2' 418 | # @@protoc_insertion_point(class_scope:CoreML.Specification.StringToInt64Map.MapEntry) 419 | )) 420 | , 421 | DESCRIPTOR = _STRINGTOINT64MAP, 422 | __module__ = 'DataStructures_pb2' 423 | # @@protoc_insertion_point(class_scope:CoreML.Specification.StringToInt64Map) 424 | )) 425 | _sym_db.RegisterMessage(StringToInt64Map) 426 | _sym_db.RegisterMessage(StringToInt64Map.MapEntry) 427 | 428 | Int64ToStringMap = _reflection.GeneratedProtocolMessageType('Int64ToStringMap', (_message.Message,), dict( 429 | 430 | MapEntry = _reflection.GeneratedProtocolMessageType('MapEntry', (_message.Message,), dict( 431 | DESCRIPTOR = _INT64TOSTRINGMAP_MAPENTRY, 432 | __module__ = 'DataStructures_pb2' 433 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Int64ToStringMap.MapEntry) 434 | )) 435 | , 436 | DESCRIPTOR = _INT64TOSTRINGMAP, 437 | __module__ = 'DataStructures_pb2' 438 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Int64ToStringMap) 439 | )) 440 | _sym_db.RegisterMessage(Int64ToStringMap) 441 | _sym_db.RegisterMessage(Int64ToStringMap.MapEntry) 442 | 443 | StringToDoubleMap = _reflection.GeneratedProtocolMessageType('StringToDoubleMap', (_message.Message,), dict( 444 | 445 | MapEntry = _reflection.GeneratedProtocolMessageType('MapEntry', (_message.Message,), dict( 446 | DESCRIPTOR = _STRINGTODOUBLEMAP_MAPENTRY, 447 | __module__ = 'DataStructures_pb2' 448 | # @@protoc_insertion_point(class_scope:CoreML.Specification.StringToDoubleMap.MapEntry) 449 | )) 450 | , 451 | DESCRIPTOR = _STRINGTODOUBLEMAP, 452 | __module__ = 'DataStructures_pb2' 453 | # @@protoc_insertion_point(class_scope:CoreML.Specification.StringToDoubleMap) 454 | )) 455 | _sym_db.RegisterMessage(StringToDoubleMap) 456 | _sym_db.RegisterMessage(StringToDoubleMap.MapEntry) 457 | 458 | Int64ToDoubleMap = _reflection.GeneratedProtocolMessageType('Int64ToDoubleMap', (_message.Message,), dict( 459 | 460 | MapEntry = _reflection.GeneratedProtocolMessageType('MapEntry', (_message.Message,), dict( 461 | DESCRIPTOR = _INT64TODOUBLEMAP_MAPENTRY, 462 | __module__ = 'DataStructures_pb2' 463 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Int64ToDoubleMap.MapEntry) 464 | )) 465 | , 466 | DESCRIPTOR = _INT64TODOUBLEMAP, 467 | __module__ = 'DataStructures_pb2' 468 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Int64ToDoubleMap) 469 | )) 470 | _sym_db.RegisterMessage(Int64ToDoubleMap) 471 | _sym_db.RegisterMessage(Int64ToDoubleMap.MapEntry) 472 | 473 | StringVector = _reflection.GeneratedProtocolMessageType('StringVector', (_message.Message,), dict( 474 | DESCRIPTOR = _STRINGVECTOR, 475 | __module__ = 'DataStructures_pb2' 476 | # @@protoc_insertion_point(class_scope:CoreML.Specification.StringVector) 477 | )) 478 | _sym_db.RegisterMessage(StringVector) 479 | 480 | Int64Vector = _reflection.GeneratedProtocolMessageType('Int64Vector', (_message.Message,), dict( 481 | DESCRIPTOR = _INT64VECTOR, 482 | __module__ = 'DataStructures_pb2' 483 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Int64Vector) 484 | )) 485 | _sym_db.RegisterMessage(Int64Vector) 486 | 487 | DoubleVector = _reflection.GeneratedProtocolMessageType('DoubleVector', (_message.Message,), dict( 488 | DESCRIPTOR = _DOUBLEVECTOR, 489 | __module__ = 'DataStructures_pb2' 490 | # @@protoc_insertion_point(class_scope:CoreML.Specification.DoubleVector) 491 | )) 492 | _sym_db.RegisterMessage(DoubleVector) 493 | 494 | 495 | DESCRIPTOR.has_options = True 496 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 497 | _STRINGTOINT64MAP_MAPENTRY.has_options = True 498 | _STRINGTOINT64MAP_MAPENTRY._options = _descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')) 499 | _INT64TOSTRINGMAP_MAPENTRY.has_options = True 500 | _INT64TOSTRINGMAP_MAPENTRY._options = _descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')) 501 | _STRINGTODOUBLEMAP_MAPENTRY.has_options = True 502 | _STRINGTODOUBLEMAP_MAPENTRY._options = _descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')) 503 | _INT64TODOUBLEMAP_MAPENTRY.has_options = True 504 | _INT64TODOUBLEMAP_MAPENTRY._options = _descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')) 505 | # @@protoc_insertion_point(module_scope) 506 | -------------------------------------------------------------------------------- /DictVectorizer_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: DictVectorizer.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import DataStructures_pb2 as DataStructures__pb2 17 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 18 | 19 | from DataStructures_pb2 import * 20 | 21 | DESCRIPTOR = _descriptor.FileDescriptor( 22 | name='DictVectorizer.proto', 23 | package='CoreML.Specification', 24 | syntax='proto3', 25 | serialized_pb=_b('\n\x14\x44ictVectorizer.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\x8f\x01\n\x0e\x44ictVectorizer\x12;\n\rstringToIndex\x18\x01 \x01(\x0b\x32\".CoreML.Specification.StringVectorH\x00\x12\x39\n\x0cint64ToIndex\x18\x02 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x00\x42\x05\n\x03MapB\x02H\x03P\x00\x62\x06proto3') 26 | , 27 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 28 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 29 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 30 | 31 | 32 | 33 | 34 | _DICTVECTORIZER = _descriptor.Descriptor( 35 | name='DictVectorizer', 36 | full_name='CoreML.Specification.DictVectorizer', 37 | filename=None, 38 | file=DESCRIPTOR, 39 | containing_type=None, 40 | fields=[ 41 | _descriptor.FieldDescriptor( 42 | name='stringToIndex', full_name='CoreML.Specification.DictVectorizer.stringToIndex', index=0, 43 | number=1, type=11, cpp_type=10, label=1, 44 | has_default_value=False, default_value=None, 45 | message_type=None, enum_type=None, containing_type=None, 46 | is_extension=False, extension_scope=None, 47 | options=None), 48 | _descriptor.FieldDescriptor( 49 | name='int64ToIndex', full_name='CoreML.Specification.DictVectorizer.int64ToIndex', index=1, 50 | number=2, type=11, cpp_type=10, label=1, 51 | has_default_value=False, default_value=None, 52 | message_type=None, enum_type=None, containing_type=None, 53 | is_extension=False, extension_scope=None, 54 | options=None), 55 | ], 56 | extensions=[ 57 | ], 58 | nested_types=[], 59 | enum_types=[ 60 | ], 61 | options=None, 62 | is_extendable=False, 63 | syntax='proto3', 64 | extension_ranges=[], 65 | oneofs=[ 66 | _descriptor.OneofDescriptor( 67 | name='Map', full_name='CoreML.Specification.DictVectorizer.Map', 68 | index=0, containing_type=None, fields=[]), 69 | ], 70 | serialized_start=69, 71 | serialized_end=212, 72 | ) 73 | 74 | _DICTVECTORIZER.fields_by_name['stringToIndex'].message_type = DataStructures__pb2._STRINGVECTOR 75 | _DICTVECTORIZER.fields_by_name['int64ToIndex'].message_type = DataStructures__pb2._INT64VECTOR 76 | _DICTVECTORIZER.oneofs_by_name['Map'].fields.append( 77 | _DICTVECTORIZER.fields_by_name['stringToIndex']) 78 | _DICTVECTORIZER.fields_by_name['stringToIndex'].containing_oneof = _DICTVECTORIZER.oneofs_by_name['Map'] 79 | _DICTVECTORIZER.oneofs_by_name['Map'].fields.append( 80 | _DICTVECTORIZER.fields_by_name['int64ToIndex']) 81 | _DICTVECTORIZER.fields_by_name['int64ToIndex'].containing_oneof = _DICTVECTORIZER.oneofs_by_name['Map'] 82 | DESCRIPTOR.message_types_by_name['DictVectorizer'] = _DICTVECTORIZER 83 | 84 | DictVectorizer = _reflection.GeneratedProtocolMessageType('DictVectorizer', (_message.Message,), dict( 85 | DESCRIPTOR = _DICTVECTORIZER, 86 | __module__ = 'DictVectorizer_pb2' 87 | # @@protoc_insertion_point(class_scope:CoreML.Specification.DictVectorizer) 88 | )) 89 | _sym_db.RegisterMessage(DictVectorizer) 90 | 91 | 92 | DESCRIPTOR.has_options = True 93 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 94 | # @@protoc_insertion_point(module_scope) 95 | -------------------------------------------------------------------------------- /FeatureTypes_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: FeatureTypes.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='FeatureTypes.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x12\x46\x65\x61tureTypes.proto\x12\x14\x43oreML.Specification\"\x12\n\x10Int64FeatureType\"\x13\n\x11\x44oubleFeatureType\"\x13\n\x11StringFeatureType\"\xc0\x01\n\x10ImageFeatureType\x12\r\n\x05width\x18\x01 \x01(\x04\x12\x0e\n\x06height\x18\x02 \x01(\x04\x12\x45\n\ncolorSpace\x18\x03 \x01(\x0e\x32\x31.CoreML.Specification.ImageFeatureType.ColorSpace\"F\n\nColorSpace\x12\x17\n\x13INVALID_COLOR_SPACE\x10\x00\x12\r\n\tGRAYSCALE\x10\n\x12\x07\n\x03RGB\x10\x14\x12\x07\n\x03\x42GR\x10\x1e\"\xc1\x01\n\x10\x41rrayFeatureType\x12\r\n\x05shape\x18\x01 \x03(\x04\x12\x46\n\x08\x64\x61taType\x18\x02 \x01(\x0e\x32\x34.CoreML.Specification.ArrayFeatureType.ArrayDataType\"V\n\rArrayDataType\x12\x1b\n\x17INVALID_ARRAY_DATA_TYPE\x10\x00\x12\r\n\x07\x46LOAT32\x10\xa0\x80\x04\x12\x0c\n\x06\x44OUBLE\x10\xc0\x80\x04\x12\x0b\n\x05INT32\x10\xa0\x80\x08\"\xa4\x01\n\x15\x44ictionaryFeatureType\x12>\n\x0cint64KeyType\x18\x01 \x01(\x0b\x32&.CoreML.Specification.Int64FeatureTypeH\x00\x12@\n\rstringKeyType\x18\x02 \x01(\x0b\x32\'.CoreML.Specification.StringFeatureTypeH\x00\x42\t\n\x07KeyType\"\xab\x03\n\x0b\x46\x65\x61tureType\x12;\n\tint64Type\x18\x01 \x01(\x0b\x32&.CoreML.Specification.Int64FeatureTypeH\x00\x12=\n\ndoubleType\x18\x02 \x01(\x0b\x32\'.CoreML.Specification.DoubleFeatureTypeH\x00\x12=\n\nstringType\x18\x03 \x01(\x0b\x32\'.CoreML.Specification.StringFeatureTypeH\x00\x12;\n\timageType\x18\x04 \x01(\x0b\x32&.CoreML.Specification.ImageFeatureTypeH\x00\x12@\n\x0emultiArrayType\x18\x05 \x01(\x0b\x32&.CoreML.Specification.ArrayFeatureTypeH\x00\x12\x45\n\x0e\x64ictionaryType\x18\x06 \x01(\x0b\x32+.CoreML.Specification.DictionaryFeatureTypeH\x00\x12\x13\n\nisOptional\x18\xe8\x07 \x01(\x08\x42\x06\n\x04TypeB\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | _IMAGEFEATURETYPE_COLORSPACE = _descriptor.EnumDescriptor( 29 | name='ColorSpace', 30 | full_name='CoreML.Specification.ImageFeatureType.ColorSpace', 31 | filename=None, 32 | file=DESCRIPTOR, 33 | values=[ 34 | _descriptor.EnumValueDescriptor( 35 | name='INVALID_COLOR_SPACE', index=0, number=0, 36 | options=None, 37 | type=None), 38 | _descriptor.EnumValueDescriptor( 39 | name='GRAYSCALE', index=1, number=10, 40 | options=None, 41 | type=None), 42 | _descriptor.EnumValueDescriptor( 43 | name='RGB', index=2, number=20, 44 | options=None, 45 | type=None), 46 | _descriptor.EnumValueDescriptor( 47 | name='BGR', index=3, number=30, 48 | options=None, 49 | type=None), 50 | ], 51 | containing_type=None, 52 | options=None, 53 | serialized_start=229, 54 | serialized_end=299, 55 | ) 56 | _sym_db.RegisterEnumDescriptor(_IMAGEFEATURETYPE_COLORSPACE) 57 | 58 | _ARRAYFEATURETYPE_ARRAYDATATYPE = _descriptor.EnumDescriptor( 59 | name='ArrayDataType', 60 | full_name='CoreML.Specification.ArrayFeatureType.ArrayDataType', 61 | filename=None, 62 | file=DESCRIPTOR, 63 | values=[ 64 | _descriptor.EnumValueDescriptor( 65 | name='INVALID_ARRAY_DATA_TYPE', index=0, number=0, 66 | options=None, 67 | type=None), 68 | _descriptor.EnumValueDescriptor( 69 | name='FLOAT32', index=1, number=65568, 70 | options=None, 71 | type=None), 72 | _descriptor.EnumValueDescriptor( 73 | name='DOUBLE', index=2, number=65600, 74 | options=None, 75 | type=None), 76 | _descriptor.EnumValueDescriptor( 77 | name='INT32', index=3, number=131104, 78 | options=None, 79 | type=None), 80 | ], 81 | containing_type=None, 82 | options=None, 83 | serialized_start=409, 84 | serialized_end=495, 85 | ) 86 | _sym_db.RegisterEnumDescriptor(_ARRAYFEATURETYPE_ARRAYDATATYPE) 87 | 88 | 89 | _INT64FEATURETYPE = _descriptor.Descriptor( 90 | name='Int64FeatureType', 91 | full_name='CoreML.Specification.Int64FeatureType', 92 | filename=None, 93 | file=DESCRIPTOR, 94 | containing_type=None, 95 | fields=[ 96 | ], 97 | extensions=[ 98 | ], 99 | nested_types=[], 100 | enum_types=[ 101 | ], 102 | options=None, 103 | is_extendable=False, 104 | syntax='proto3', 105 | extension_ranges=[], 106 | oneofs=[ 107 | ], 108 | serialized_start=44, 109 | serialized_end=62, 110 | ) 111 | 112 | 113 | _DOUBLEFEATURETYPE = _descriptor.Descriptor( 114 | name='DoubleFeatureType', 115 | full_name='CoreML.Specification.DoubleFeatureType', 116 | filename=None, 117 | file=DESCRIPTOR, 118 | containing_type=None, 119 | fields=[ 120 | ], 121 | extensions=[ 122 | ], 123 | nested_types=[], 124 | enum_types=[ 125 | ], 126 | options=None, 127 | is_extendable=False, 128 | syntax='proto3', 129 | extension_ranges=[], 130 | oneofs=[ 131 | ], 132 | serialized_start=64, 133 | serialized_end=83, 134 | ) 135 | 136 | 137 | _STRINGFEATURETYPE = _descriptor.Descriptor( 138 | name='StringFeatureType', 139 | full_name='CoreML.Specification.StringFeatureType', 140 | filename=None, 141 | file=DESCRIPTOR, 142 | containing_type=None, 143 | fields=[ 144 | ], 145 | extensions=[ 146 | ], 147 | nested_types=[], 148 | enum_types=[ 149 | ], 150 | options=None, 151 | is_extendable=False, 152 | syntax='proto3', 153 | extension_ranges=[], 154 | oneofs=[ 155 | ], 156 | serialized_start=85, 157 | serialized_end=104, 158 | ) 159 | 160 | 161 | _IMAGEFEATURETYPE = _descriptor.Descriptor( 162 | name='ImageFeatureType', 163 | full_name='CoreML.Specification.ImageFeatureType', 164 | filename=None, 165 | file=DESCRIPTOR, 166 | containing_type=None, 167 | fields=[ 168 | _descriptor.FieldDescriptor( 169 | name='width', full_name='CoreML.Specification.ImageFeatureType.width', index=0, 170 | number=1, type=4, cpp_type=4, label=1, 171 | has_default_value=False, default_value=0, 172 | message_type=None, enum_type=None, containing_type=None, 173 | is_extension=False, extension_scope=None, 174 | options=None), 175 | _descriptor.FieldDescriptor( 176 | name='height', full_name='CoreML.Specification.ImageFeatureType.height', index=1, 177 | number=2, type=4, cpp_type=4, label=1, 178 | has_default_value=False, default_value=0, 179 | message_type=None, enum_type=None, containing_type=None, 180 | is_extension=False, extension_scope=None, 181 | options=None), 182 | _descriptor.FieldDescriptor( 183 | name='colorSpace', full_name='CoreML.Specification.ImageFeatureType.colorSpace', index=2, 184 | number=3, type=14, cpp_type=8, label=1, 185 | has_default_value=False, default_value=0, 186 | message_type=None, enum_type=None, containing_type=None, 187 | is_extension=False, extension_scope=None, 188 | options=None), 189 | ], 190 | extensions=[ 191 | ], 192 | nested_types=[], 193 | enum_types=[ 194 | _IMAGEFEATURETYPE_COLORSPACE, 195 | ], 196 | options=None, 197 | is_extendable=False, 198 | syntax='proto3', 199 | extension_ranges=[], 200 | oneofs=[ 201 | ], 202 | serialized_start=107, 203 | serialized_end=299, 204 | ) 205 | 206 | 207 | _ARRAYFEATURETYPE = _descriptor.Descriptor( 208 | name='ArrayFeatureType', 209 | full_name='CoreML.Specification.ArrayFeatureType', 210 | filename=None, 211 | file=DESCRIPTOR, 212 | containing_type=None, 213 | fields=[ 214 | _descriptor.FieldDescriptor( 215 | name='shape', full_name='CoreML.Specification.ArrayFeatureType.shape', index=0, 216 | number=1, type=4, cpp_type=4, label=3, 217 | has_default_value=False, default_value=[], 218 | message_type=None, enum_type=None, containing_type=None, 219 | is_extension=False, extension_scope=None, 220 | options=None), 221 | _descriptor.FieldDescriptor( 222 | name='dataType', full_name='CoreML.Specification.ArrayFeatureType.dataType', index=1, 223 | number=2, type=14, cpp_type=8, label=1, 224 | has_default_value=False, default_value=0, 225 | message_type=None, enum_type=None, containing_type=None, 226 | is_extension=False, extension_scope=None, 227 | options=None), 228 | ], 229 | extensions=[ 230 | ], 231 | nested_types=[], 232 | enum_types=[ 233 | _ARRAYFEATURETYPE_ARRAYDATATYPE, 234 | ], 235 | options=None, 236 | is_extendable=False, 237 | syntax='proto3', 238 | extension_ranges=[], 239 | oneofs=[ 240 | ], 241 | serialized_start=302, 242 | serialized_end=495, 243 | ) 244 | 245 | 246 | _DICTIONARYFEATURETYPE = _descriptor.Descriptor( 247 | name='DictionaryFeatureType', 248 | full_name='CoreML.Specification.DictionaryFeatureType', 249 | filename=None, 250 | file=DESCRIPTOR, 251 | containing_type=None, 252 | fields=[ 253 | _descriptor.FieldDescriptor( 254 | name='int64KeyType', full_name='CoreML.Specification.DictionaryFeatureType.int64KeyType', index=0, 255 | number=1, type=11, cpp_type=10, label=1, 256 | has_default_value=False, default_value=None, 257 | message_type=None, enum_type=None, containing_type=None, 258 | is_extension=False, extension_scope=None, 259 | options=None), 260 | _descriptor.FieldDescriptor( 261 | name='stringKeyType', full_name='CoreML.Specification.DictionaryFeatureType.stringKeyType', index=1, 262 | number=2, type=11, cpp_type=10, label=1, 263 | has_default_value=False, default_value=None, 264 | message_type=None, enum_type=None, containing_type=None, 265 | is_extension=False, extension_scope=None, 266 | options=None), 267 | ], 268 | extensions=[ 269 | ], 270 | nested_types=[], 271 | enum_types=[ 272 | ], 273 | options=None, 274 | is_extendable=False, 275 | syntax='proto3', 276 | extension_ranges=[], 277 | oneofs=[ 278 | _descriptor.OneofDescriptor( 279 | name='KeyType', full_name='CoreML.Specification.DictionaryFeatureType.KeyType', 280 | index=0, containing_type=None, fields=[]), 281 | ], 282 | serialized_start=498, 283 | serialized_end=662, 284 | ) 285 | 286 | 287 | _FEATURETYPE = _descriptor.Descriptor( 288 | name='FeatureType', 289 | full_name='CoreML.Specification.FeatureType', 290 | filename=None, 291 | file=DESCRIPTOR, 292 | containing_type=None, 293 | fields=[ 294 | _descriptor.FieldDescriptor( 295 | name='int64Type', full_name='CoreML.Specification.FeatureType.int64Type', index=0, 296 | number=1, type=11, cpp_type=10, label=1, 297 | has_default_value=False, default_value=None, 298 | message_type=None, enum_type=None, containing_type=None, 299 | is_extension=False, extension_scope=None, 300 | options=None), 301 | _descriptor.FieldDescriptor( 302 | name='doubleType', full_name='CoreML.Specification.FeatureType.doubleType', index=1, 303 | number=2, type=11, cpp_type=10, label=1, 304 | has_default_value=False, default_value=None, 305 | message_type=None, enum_type=None, containing_type=None, 306 | is_extension=False, extension_scope=None, 307 | options=None), 308 | _descriptor.FieldDescriptor( 309 | name='stringType', full_name='CoreML.Specification.FeatureType.stringType', index=2, 310 | number=3, type=11, cpp_type=10, label=1, 311 | has_default_value=False, default_value=None, 312 | message_type=None, enum_type=None, containing_type=None, 313 | is_extension=False, extension_scope=None, 314 | options=None), 315 | _descriptor.FieldDescriptor( 316 | name='imageType', full_name='CoreML.Specification.FeatureType.imageType', index=3, 317 | number=4, type=11, cpp_type=10, label=1, 318 | has_default_value=False, default_value=None, 319 | message_type=None, enum_type=None, containing_type=None, 320 | is_extension=False, extension_scope=None, 321 | options=None), 322 | _descriptor.FieldDescriptor( 323 | name='multiArrayType', full_name='CoreML.Specification.FeatureType.multiArrayType', index=4, 324 | number=5, type=11, cpp_type=10, label=1, 325 | has_default_value=False, default_value=None, 326 | message_type=None, enum_type=None, containing_type=None, 327 | is_extension=False, extension_scope=None, 328 | options=None), 329 | _descriptor.FieldDescriptor( 330 | name='dictionaryType', full_name='CoreML.Specification.FeatureType.dictionaryType', index=5, 331 | number=6, type=11, cpp_type=10, label=1, 332 | has_default_value=False, default_value=None, 333 | message_type=None, enum_type=None, containing_type=None, 334 | is_extension=False, extension_scope=None, 335 | options=None), 336 | _descriptor.FieldDescriptor( 337 | name='isOptional', full_name='CoreML.Specification.FeatureType.isOptional', index=6, 338 | number=1000, type=8, cpp_type=7, label=1, 339 | has_default_value=False, default_value=False, 340 | message_type=None, enum_type=None, containing_type=None, 341 | is_extension=False, extension_scope=None, 342 | options=None), 343 | ], 344 | extensions=[ 345 | ], 346 | nested_types=[], 347 | enum_types=[ 348 | ], 349 | options=None, 350 | is_extendable=False, 351 | syntax='proto3', 352 | extension_ranges=[], 353 | oneofs=[ 354 | _descriptor.OneofDescriptor( 355 | name='Type', full_name='CoreML.Specification.FeatureType.Type', 356 | index=0, containing_type=None, fields=[]), 357 | ], 358 | serialized_start=665, 359 | serialized_end=1092, 360 | ) 361 | 362 | _IMAGEFEATURETYPE.fields_by_name['colorSpace'].enum_type = _IMAGEFEATURETYPE_COLORSPACE 363 | _IMAGEFEATURETYPE_COLORSPACE.containing_type = _IMAGEFEATURETYPE 364 | _ARRAYFEATURETYPE.fields_by_name['dataType'].enum_type = _ARRAYFEATURETYPE_ARRAYDATATYPE 365 | _ARRAYFEATURETYPE_ARRAYDATATYPE.containing_type = _ARRAYFEATURETYPE 366 | _DICTIONARYFEATURETYPE.fields_by_name['int64KeyType'].message_type = _INT64FEATURETYPE 367 | _DICTIONARYFEATURETYPE.fields_by_name['stringKeyType'].message_type = _STRINGFEATURETYPE 368 | _DICTIONARYFEATURETYPE.oneofs_by_name['KeyType'].fields.append( 369 | _DICTIONARYFEATURETYPE.fields_by_name['int64KeyType']) 370 | _DICTIONARYFEATURETYPE.fields_by_name['int64KeyType'].containing_oneof = _DICTIONARYFEATURETYPE.oneofs_by_name['KeyType'] 371 | _DICTIONARYFEATURETYPE.oneofs_by_name['KeyType'].fields.append( 372 | _DICTIONARYFEATURETYPE.fields_by_name['stringKeyType']) 373 | _DICTIONARYFEATURETYPE.fields_by_name['stringKeyType'].containing_oneof = _DICTIONARYFEATURETYPE.oneofs_by_name['KeyType'] 374 | _FEATURETYPE.fields_by_name['int64Type'].message_type = _INT64FEATURETYPE 375 | _FEATURETYPE.fields_by_name['doubleType'].message_type = _DOUBLEFEATURETYPE 376 | _FEATURETYPE.fields_by_name['stringType'].message_type = _STRINGFEATURETYPE 377 | _FEATURETYPE.fields_by_name['imageType'].message_type = _IMAGEFEATURETYPE 378 | _FEATURETYPE.fields_by_name['multiArrayType'].message_type = _ARRAYFEATURETYPE 379 | _FEATURETYPE.fields_by_name['dictionaryType'].message_type = _DICTIONARYFEATURETYPE 380 | _FEATURETYPE.oneofs_by_name['Type'].fields.append( 381 | _FEATURETYPE.fields_by_name['int64Type']) 382 | _FEATURETYPE.fields_by_name['int64Type'].containing_oneof = _FEATURETYPE.oneofs_by_name['Type'] 383 | _FEATURETYPE.oneofs_by_name['Type'].fields.append( 384 | _FEATURETYPE.fields_by_name['doubleType']) 385 | _FEATURETYPE.fields_by_name['doubleType'].containing_oneof = _FEATURETYPE.oneofs_by_name['Type'] 386 | _FEATURETYPE.oneofs_by_name['Type'].fields.append( 387 | _FEATURETYPE.fields_by_name['stringType']) 388 | _FEATURETYPE.fields_by_name['stringType'].containing_oneof = _FEATURETYPE.oneofs_by_name['Type'] 389 | _FEATURETYPE.oneofs_by_name['Type'].fields.append( 390 | _FEATURETYPE.fields_by_name['imageType']) 391 | _FEATURETYPE.fields_by_name['imageType'].containing_oneof = _FEATURETYPE.oneofs_by_name['Type'] 392 | _FEATURETYPE.oneofs_by_name['Type'].fields.append( 393 | _FEATURETYPE.fields_by_name['multiArrayType']) 394 | _FEATURETYPE.fields_by_name['multiArrayType'].containing_oneof = _FEATURETYPE.oneofs_by_name['Type'] 395 | _FEATURETYPE.oneofs_by_name['Type'].fields.append( 396 | _FEATURETYPE.fields_by_name['dictionaryType']) 397 | _FEATURETYPE.fields_by_name['dictionaryType'].containing_oneof = _FEATURETYPE.oneofs_by_name['Type'] 398 | DESCRIPTOR.message_types_by_name['Int64FeatureType'] = _INT64FEATURETYPE 399 | DESCRIPTOR.message_types_by_name['DoubleFeatureType'] = _DOUBLEFEATURETYPE 400 | DESCRIPTOR.message_types_by_name['StringFeatureType'] = _STRINGFEATURETYPE 401 | DESCRIPTOR.message_types_by_name['ImageFeatureType'] = _IMAGEFEATURETYPE 402 | DESCRIPTOR.message_types_by_name['ArrayFeatureType'] = _ARRAYFEATURETYPE 403 | DESCRIPTOR.message_types_by_name['DictionaryFeatureType'] = _DICTIONARYFEATURETYPE 404 | DESCRIPTOR.message_types_by_name['FeatureType'] = _FEATURETYPE 405 | 406 | Int64FeatureType = _reflection.GeneratedProtocolMessageType('Int64FeatureType', (_message.Message,), dict( 407 | DESCRIPTOR = _INT64FEATURETYPE, 408 | __module__ = 'FeatureTypes_pb2' 409 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Int64FeatureType) 410 | )) 411 | _sym_db.RegisterMessage(Int64FeatureType) 412 | 413 | DoubleFeatureType = _reflection.GeneratedProtocolMessageType('DoubleFeatureType', (_message.Message,), dict( 414 | DESCRIPTOR = _DOUBLEFEATURETYPE, 415 | __module__ = 'FeatureTypes_pb2' 416 | # @@protoc_insertion_point(class_scope:CoreML.Specification.DoubleFeatureType) 417 | )) 418 | _sym_db.RegisterMessage(DoubleFeatureType) 419 | 420 | StringFeatureType = _reflection.GeneratedProtocolMessageType('StringFeatureType', (_message.Message,), dict( 421 | DESCRIPTOR = _STRINGFEATURETYPE, 422 | __module__ = 'FeatureTypes_pb2' 423 | # @@protoc_insertion_point(class_scope:CoreML.Specification.StringFeatureType) 424 | )) 425 | _sym_db.RegisterMessage(StringFeatureType) 426 | 427 | ImageFeatureType = _reflection.GeneratedProtocolMessageType('ImageFeatureType', (_message.Message,), dict( 428 | DESCRIPTOR = _IMAGEFEATURETYPE, 429 | __module__ = 'FeatureTypes_pb2' 430 | # @@protoc_insertion_point(class_scope:CoreML.Specification.ImageFeatureType) 431 | )) 432 | _sym_db.RegisterMessage(ImageFeatureType) 433 | 434 | ArrayFeatureType = _reflection.GeneratedProtocolMessageType('ArrayFeatureType', (_message.Message,), dict( 435 | DESCRIPTOR = _ARRAYFEATURETYPE, 436 | __module__ = 'FeatureTypes_pb2' 437 | # @@protoc_insertion_point(class_scope:CoreML.Specification.ArrayFeatureType) 438 | )) 439 | _sym_db.RegisterMessage(ArrayFeatureType) 440 | 441 | DictionaryFeatureType = _reflection.GeneratedProtocolMessageType('DictionaryFeatureType', (_message.Message,), dict( 442 | DESCRIPTOR = _DICTIONARYFEATURETYPE, 443 | __module__ = 'FeatureTypes_pb2' 444 | # @@protoc_insertion_point(class_scope:CoreML.Specification.DictionaryFeatureType) 445 | )) 446 | _sym_db.RegisterMessage(DictionaryFeatureType) 447 | 448 | FeatureType = _reflection.GeneratedProtocolMessageType('FeatureType', (_message.Message,), dict( 449 | DESCRIPTOR = _FEATURETYPE, 450 | __module__ = 'FeatureTypes_pb2' 451 | # @@protoc_insertion_point(class_scope:CoreML.Specification.FeatureType) 452 | )) 453 | _sym_db.RegisterMessage(FeatureType) 454 | 455 | 456 | DESCRIPTOR.has_options = True 457 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 458 | # @@protoc_insertion_point(module_scope) 459 | -------------------------------------------------------------------------------- /FeatureVectorizer_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: FeatureVectorizer.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='FeatureVectorizer.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x17\x46\x65\x61tureVectorizer.proto\x12\x14\x43oreML.Specification\"\x98\x01\n\x11\x46\x65\x61tureVectorizer\x12\x46\n\tinputList\x18\x01 \x03(\x0b\x32\x33.CoreML.Specification.FeatureVectorizer.InputColumn\x1a;\n\x0bInputColumn\x12\x13\n\x0binputColumn\x18\x01 \x01(\t\x12\x17\n\x0finputDimensions\x18\x02 \x01(\x04\x42\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | 29 | _FEATUREVECTORIZER_INPUTCOLUMN = _descriptor.Descriptor( 30 | name='InputColumn', 31 | full_name='CoreML.Specification.FeatureVectorizer.InputColumn', 32 | filename=None, 33 | file=DESCRIPTOR, 34 | containing_type=None, 35 | fields=[ 36 | _descriptor.FieldDescriptor( 37 | name='inputColumn', full_name='CoreML.Specification.FeatureVectorizer.InputColumn.inputColumn', index=0, 38 | number=1, type=9, cpp_type=9, label=1, 39 | has_default_value=False, default_value=_b("").decode('utf-8'), 40 | message_type=None, enum_type=None, containing_type=None, 41 | is_extension=False, extension_scope=None, 42 | options=None), 43 | _descriptor.FieldDescriptor( 44 | name='inputDimensions', full_name='CoreML.Specification.FeatureVectorizer.InputColumn.inputDimensions', index=1, 45 | number=2, type=4, cpp_type=4, label=1, 46 | has_default_value=False, default_value=0, 47 | message_type=None, enum_type=None, containing_type=None, 48 | is_extension=False, extension_scope=None, 49 | options=None), 50 | ], 51 | extensions=[ 52 | ], 53 | nested_types=[], 54 | enum_types=[ 55 | ], 56 | options=None, 57 | is_extendable=False, 58 | syntax='proto3', 59 | extension_ranges=[], 60 | oneofs=[ 61 | ], 62 | serialized_start=143, 63 | serialized_end=202, 64 | ) 65 | 66 | _FEATUREVECTORIZER = _descriptor.Descriptor( 67 | name='FeatureVectorizer', 68 | full_name='CoreML.Specification.FeatureVectorizer', 69 | filename=None, 70 | file=DESCRIPTOR, 71 | containing_type=None, 72 | fields=[ 73 | _descriptor.FieldDescriptor( 74 | name='inputList', full_name='CoreML.Specification.FeatureVectorizer.inputList', index=0, 75 | number=1, type=11, cpp_type=10, label=3, 76 | has_default_value=False, default_value=[], 77 | message_type=None, enum_type=None, containing_type=None, 78 | is_extension=False, extension_scope=None, 79 | options=None), 80 | ], 81 | extensions=[ 82 | ], 83 | nested_types=[_FEATUREVECTORIZER_INPUTCOLUMN, ], 84 | enum_types=[ 85 | ], 86 | options=None, 87 | is_extendable=False, 88 | syntax='proto3', 89 | extension_ranges=[], 90 | oneofs=[ 91 | ], 92 | serialized_start=50, 93 | serialized_end=202, 94 | ) 95 | 96 | _FEATUREVECTORIZER_INPUTCOLUMN.containing_type = _FEATUREVECTORIZER 97 | _FEATUREVECTORIZER.fields_by_name['inputList'].message_type = _FEATUREVECTORIZER_INPUTCOLUMN 98 | DESCRIPTOR.message_types_by_name['FeatureVectorizer'] = _FEATUREVECTORIZER 99 | 100 | FeatureVectorizer = _reflection.GeneratedProtocolMessageType('FeatureVectorizer', (_message.Message,), dict( 101 | 102 | InputColumn = _reflection.GeneratedProtocolMessageType('InputColumn', (_message.Message,), dict( 103 | DESCRIPTOR = _FEATUREVECTORIZER_INPUTCOLUMN, 104 | __module__ = 'FeatureVectorizer_pb2' 105 | # @@protoc_insertion_point(class_scope:CoreML.Specification.FeatureVectorizer.InputColumn) 106 | )) 107 | , 108 | DESCRIPTOR = _FEATUREVECTORIZER, 109 | __module__ = 'FeatureVectorizer_pb2' 110 | # @@protoc_insertion_point(class_scope:CoreML.Specification.FeatureVectorizer) 111 | )) 112 | _sym_db.RegisterMessage(FeatureVectorizer) 113 | _sym_db.RegisterMessage(FeatureVectorizer.InputColumn) 114 | 115 | 116 | DESCRIPTOR.has_options = True 117 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 118 | # @@protoc_insertion_point(module_scope) 119 | -------------------------------------------------------------------------------- /GLMClassifier_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: GLMClassifier.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import DataStructures_pb2 as DataStructures__pb2 17 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 18 | 19 | from DataStructures_pb2 import * 20 | 21 | DESCRIPTOR = _descriptor.FileDescriptor( 22 | name='GLMClassifier.proto', 23 | package='CoreML.Specification', 24 | syntax='proto3', 25 | serialized_pb=_b('\n\x13GLMClassifier.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\x9c\x04\n\rGLMClassifier\x12@\n\x07weights\x18\x01 \x03(\x0b\x32/.CoreML.Specification.GLMClassifier.DoubleArray\x12\x0e\n\x06offset\x18\x02 \x03(\x01\x12\\\n\x17postEvaluationTransform\x18\x03 \x01(\x0e\x32;.CoreML.Specification.GLMClassifier.PostEvaluationTransform\x12H\n\rclassEncoding\x18\x04 \x01(\x0e\x32\x31.CoreML.Specification.GLMClassifier.ClassEncoding\x12?\n\x11stringClassLabels\x18\x64 \x01(\x0b\x32\".CoreML.Specification.StringVectorH\x00\x12=\n\x10int64ClassLabels\x18\x65 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x00\x1a\x1c\n\x0b\x44oubleArray\x12\r\n\x05value\x18\x01 \x03(\x01\"0\n\x17PostEvaluationTransform\x12\t\n\x05Logit\x10\x00\x12\n\n\x06Probit\x10\x01\"2\n\rClassEncoding\x12\x12\n\x0eReferenceClass\x10\x00\x12\r\n\tOneVsRest\x10\x01\x42\r\n\x0b\x43lassLabelsB\x02H\x03P\x00\x62\x06proto3') 26 | , 27 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 28 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 29 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 30 | 31 | 32 | 33 | _GLMCLASSIFIER_POSTEVALUATIONTRANSFORM = _descriptor.EnumDescriptor( 34 | name='PostEvaluationTransform', 35 | full_name='CoreML.Specification.GLMClassifier.PostEvaluationTransform', 36 | filename=None, 37 | file=DESCRIPTOR, 38 | values=[ 39 | _descriptor.EnumValueDescriptor( 40 | name='Logit', index=0, number=0, 41 | options=None, 42 | type=None), 43 | _descriptor.EnumValueDescriptor( 44 | name='Probit', index=1, number=1, 45 | options=None, 46 | type=None), 47 | ], 48 | containing_type=None, 49 | options=None, 50 | serialized_start=493, 51 | serialized_end=541, 52 | ) 53 | _sym_db.RegisterEnumDescriptor(_GLMCLASSIFIER_POSTEVALUATIONTRANSFORM) 54 | 55 | _GLMCLASSIFIER_CLASSENCODING = _descriptor.EnumDescriptor( 56 | name='ClassEncoding', 57 | full_name='CoreML.Specification.GLMClassifier.ClassEncoding', 58 | filename=None, 59 | file=DESCRIPTOR, 60 | values=[ 61 | _descriptor.EnumValueDescriptor( 62 | name='ReferenceClass', index=0, number=0, 63 | options=None, 64 | type=None), 65 | _descriptor.EnumValueDescriptor( 66 | name='OneVsRest', index=1, number=1, 67 | options=None, 68 | type=None), 69 | ], 70 | containing_type=None, 71 | options=None, 72 | serialized_start=543, 73 | serialized_end=593, 74 | ) 75 | _sym_db.RegisterEnumDescriptor(_GLMCLASSIFIER_CLASSENCODING) 76 | 77 | 78 | _GLMCLASSIFIER_DOUBLEARRAY = _descriptor.Descriptor( 79 | name='DoubleArray', 80 | full_name='CoreML.Specification.GLMClassifier.DoubleArray', 81 | filename=None, 82 | file=DESCRIPTOR, 83 | containing_type=None, 84 | fields=[ 85 | _descriptor.FieldDescriptor( 86 | name='value', full_name='CoreML.Specification.GLMClassifier.DoubleArray.value', index=0, 87 | number=1, type=1, cpp_type=5, label=3, 88 | has_default_value=False, default_value=[], 89 | message_type=None, enum_type=None, containing_type=None, 90 | is_extension=False, extension_scope=None, 91 | options=None), 92 | ], 93 | extensions=[ 94 | ], 95 | nested_types=[], 96 | enum_types=[ 97 | ], 98 | options=None, 99 | is_extendable=False, 100 | syntax='proto3', 101 | extension_ranges=[], 102 | oneofs=[ 103 | ], 104 | serialized_start=463, 105 | serialized_end=491, 106 | ) 107 | 108 | _GLMCLASSIFIER = _descriptor.Descriptor( 109 | name='GLMClassifier', 110 | full_name='CoreML.Specification.GLMClassifier', 111 | filename=None, 112 | file=DESCRIPTOR, 113 | containing_type=None, 114 | fields=[ 115 | _descriptor.FieldDescriptor( 116 | name='weights', full_name='CoreML.Specification.GLMClassifier.weights', index=0, 117 | number=1, type=11, cpp_type=10, label=3, 118 | has_default_value=False, default_value=[], 119 | message_type=None, enum_type=None, containing_type=None, 120 | is_extension=False, extension_scope=None, 121 | options=None), 122 | _descriptor.FieldDescriptor( 123 | name='offset', full_name='CoreML.Specification.GLMClassifier.offset', index=1, 124 | number=2, type=1, cpp_type=5, label=3, 125 | has_default_value=False, default_value=[], 126 | message_type=None, enum_type=None, containing_type=None, 127 | is_extension=False, extension_scope=None, 128 | options=None), 129 | _descriptor.FieldDescriptor( 130 | name='postEvaluationTransform', full_name='CoreML.Specification.GLMClassifier.postEvaluationTransform', index=2, 131 | number=3, type=14, cpp_type=8, label=1, 132 | has_default_value=False, default_value=0, 133 | message_type=None, enum_type=None, containing_type=None, 134 | is_extension=False, extension_scope=None, 135 | options=None), 136 | _descriptor.FieldDescriptor( 137 | name='classEncoding', full_name='CoreML.Specification.GLMClassifier.classEncoding', index=3, 138 | number=4, type=14, cpp_type=8, label=1, 139 | has_default_value=False, default_value=0, 140 | message_type=None, enum_type=None, containing_type=None, 141 | is_extension=False, extension_scope=None, 142 | options=None), 143 | _descriptor.FieldDescriptor( 144 | name='stringClassLabels', full_name='CoreML.Specification.GLMClassifier.stringClassLabels', index=4, 145 | number=100, type=11, cpp_type=10, label=1, 146 | has_default_value=False, default_value=None, 147 | message_type=None, enum_type=None, containing_type=None, 148 | is_extension=False, extension_scope=None, 149 | options=None), 150 | _descriptor.FieldDescriptor( 151 | name='int64ClassLabels', full_name='CoreML.Specification.GLMClassifier.int64ClassLabels', index=5, 152 | number=101, type=11, cpp_type=10, label=1, 153 | has_default_value=False, default_value=None, 154 | message_type=None, enum_type=None, containing_type=None, 155 | is_extension=False, extension_scope=None, 156 | options=None), 157 | ], 158 | extensions=[ 159 | ], 160 | nested_types=[_GLMCLASSIFIER_DOUBLEARRAY, ], 161 | enum_types=[ 162 | _GLMCLASSIFIER_POSTEVALUATIONTRANSFORM, 163 | _GLMCLASSIFIER_CLASSENCODING, 164 | ], 165 | options=None, 166 | is_extendable=False, 167 | syntax='proto3', 168 | extension_ranges=[], 169 | oneofs=[ 170 | _descriptor.OneofDescriptor( 171 | name='ClassLabels', full_name='CoreML.Specification.GLMClassifier.ClassLabels', 172 | index=0, containing_type=None, fields=[]), 173 | ], 174 | serialized_start=68, 175 | serialized_end=608, 176 | ) 177 | 178 | _GLMCLASSIFIER_DOUBLEARRAY.containing_type = _GLMCLASSIFIER 179 | _GLMCLASSIFIER.fields_by_name['weights'].message_type = _GLMCLASSIFIER_DOUBLEARRAY 180 | _GLMCLASSIFIER.fields_by_name['postEvaluationTransform'].enum_type = _GLMCLASSIFIER_POSTEVALUATIONTRANSFORM 181 | _GLMCLASSIFIER.fields_by_name['classEncoding'].enum_type = _GLMCLASSIFIER_CLASSENCODING 182 | _GLMCLASSIFIER.fields_by_name['stringClassLabels'].message_type = DataStructures__pb2._STRINGVECTOR 183 | _GLMCLASSIFIER.fields_by_name['int64ClassLabels'].message_type = DataStructures__pb2._INT64VECTOR 184 | _GLMCLASSIFIER_POSTEVALUATIONTRANSFORM.containing_type = _GLMCLASSIFIER 185 | _GLMCLASSIFIER_CLASSENCODING.containing_type = _GLMCLASSIFIER 186 | _GLMCLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 187 | _GLMCLASSIFIER.fields_by_name['stringClassLabels']) 188 | _GLMCLASSIFIER.fields_by_name['stringClassLabels'].containing_oneof = _GLMCLASSIFIER.oneofs_by_name['ClassLabels'] 189 | _GLMCLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 190 | _GLMCLASSIFIER.fields_by_name['int64ClassLabels']) 191 | _GLMCLASSIFIER.fields_by_name['int64ClassLabels'].containing_oneof = _GLMCLASSIFIER.oneofs_by_name['ClassLabels'] 192 | DESCRIPTOR.message_types_by_name['GLMClassifier'] = _GLMCLASSIFIER 193 | 194 | GLMClassifier = _reflection.GeneratedProtocolMessageType('GLMClassifier', (_message.Message,), dict( 195 | 196 | DoubleArray = _reflection.GeneratedProtocolMessageType('DoubleArray', (_message.Message,), dict( 197 | DESCRIPTOR = _GLMCLASSIFIER_DOUBLEARRAY, 198 | __module__ = 'GLMClassifier_pb2' 199 | # @@protoc_insertion_point(class_scope:CoreML.Specification.GLMClassifier.DoubleArray) 200 | )) 201 | , 202 | DESCRIPTOR = _GLMCLASSIFIER, 203 | __module__ = 'GLMClassifier_pb2' 204 | # @@protoc_insertion_point(class_scope:CoreML.Specification.GLMClassifier) 205 | )) 206 | _sym_db.RegisterMessage(GLMClassifier) 207 | _sym_db.RegisterMessage(GLMClassifier.DoubleArray) 208 | 209 | 210 | DESCRIPTOR.has_options = True 211 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 212 | # @@protoc_insertion_point(module_scope) 213 | -------------------------------------------------------------------------------- /GLMRegressor_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: GLMRegressor.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='GLMRegressor.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x12GLMRegressor.proto\x12\x14\x43oreML.Specification\"\x9d\x02\n\x0cGLMRegressor\x12?\n\x07weights\x18\x01 \x03(\x0b\x32..CoreML.Specification.GLMRegressor.DoubleArray\x12\x0e\n\x06offset\x18\x02 \x03(\x01\x12[\n\x17postEvaluationTransform\x18\x03 \x01(\x0e\x32:.CoreML.Specification.GLMRegressor.PostEvaluationTransform\x1a\x1c\n\x0b\x44oubleArray\x12\r\n\x05value\x18\x01 \x03(\x01\"A\n\x17PostEvaluationTransform\x12\x0f\n\x0bNoTransform\x10\x00\x12\t\n\x05Logit\x10\x01\x12\n\n\x06Probit\x10\x02\x42\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | _GLMREGRESSOR_POSTEVALUATIONTRANSFORM = _descriptor.EnumDescriptor( 29 | name='PostEvaluationTransform', 30 | full_name='CoreML.Specification.GLMRegressor.PostEvaluationTransform', 31 | filename=None, 32 | file=DESCRIPTOR, 33 | values=[ 34 | _descriptor.EnumValueDescriptor( 35 | name='NoTransform', index=0, number=0, 36 | options=None, 37 | type=None), 38 | _descriptor.EnumValueDescriptor( 39 | name='Logit', index=1, number=1, 40 | options=None, 41 | type=None), 42 | _descriptor.EnumValueDescriptor( 43 | name='Probit', index=2, number=2, 44 | options=None, 45 | type=None), 46 | ], 47 | containing_type=None, 48 | options=None, 49 | serialized_start=265, 50 | serialized_end=330, 51 | ) 52 | _sym_db.RegisterEnumDescriptor(_GLMREGRESSOR_POSTEVALUATIONTRANSFORM) 53 | 54 | 55 | _GLMREGRESSOR_DOUBLEARRAY = _descriptor.Descriptor( 56 | name='DoubleArray', 57 | full_name='CoreML.Specification.GLMRegressor.DoubleArray', 58 | filename=None, 59 | file=DESCRIPTOR, 60 | containing_type=None, 61 | fields=[ 62 | _descriptor.FieldDescriptor( 63 | name='value', full_name='CoreML.Specification.GLMRegressor.DoubleArray.value', index=0, 64 | number=1, type=1, cpp_type=5, label=3, 65 | has_default_value=False, default_value=[], 66 | message_type=None, enum_type=None, containing_type=None, 67 | is_extension=False, extension_scope=None, 68 | options=None), 69 | ], 70 | extensions=[ 71 | ], 72 | nested_types=[], 73 | enum_types=[ 74 | ], 75 | options=None, 76 | is_extendable=False, 77 | syntax='proto3', 78 | extension_ranges=[], 79 | oneofs=[ 80 | ], 81 | serialized_start=235, 82 | serialized_end=263, 83 | ) 84 | 85 | _GLMREGRESSOR = _descriptor.Descriptor( 86 | name='GLMRegressor', 87 | full_name='CoreML.Specification.GLMRegressor', 88 | filename=None, 89 | file=DESCRIPTOR, 90 | containing_type=None, 91 | fields=[ 92 | _descriptor.FieldDescriptor( 93 | name='weights', full_name='CoreML.Specification.GLMRegressor.weights', index=0, 94 | number=1, type=11, cpp_type=10, label=3, 95 | has_default_value=False, default_value=[], 96 | message_type=None, enum_type=None, containing_type=None, 97 | is_extension=False, extension_scope=None, 98 | options=None), 99 | _descriptor.FieldDescriptor( 100 | name='offset', full_name='CoreML.Specification.GLMRegressor.offset', index=1, 101 | number=2, type=1, cpp_type=5, label=3, 102 | has_default_value=False, default_value=[], 103 | message_type=None, enum_type=None, containing_type=None, 104 | is_extension=False, extension_scope=None, 105 | options=None), 106 | _descriptor.FieldDescriptor( 107 | name='postEvaluationTransform', full_name='CoreML.Specification.GLMRegressor.postEvaluationTransform', index=2, 108 | number=3, type=14, cpp_type=8, label=1, 109 | has_default_value=False, default_value=0, 110 | message_type=None, enum_type=None, containing_type=None, 111 | is_extension=False, extension_scope=None, 112 | options=None), 113 | ], 114 | extensions=[ 115 | ], 116 | nested_types=[_GLMREGRESSOR_DOUBLEARRAY, ], 117 | enum_types=[ 118 | _GLMREGRESSOR_POSTEVALUATIONTRANSFORM, 119 | ], 120 | options=None, 121 | is_extendable=False, 122 | syntax='proto3', 123 | extension_ranges=[], 124 | oneofs=[ 125 | ], 126 | serialized_start=45, 127 | serialized_end=330, 128 | ) 129 | 130 | _GLMREGRESSOR_DOUBLEARRAY.containing_type = _GLMREGRESSOR 131 | _GLMREGRESSOR.fields_by_name['weights'].message_type = _GLMREGRESSOR_DOUBLEARRAY 132 | _GLMREGRESSOR.fields_by_name['postEvaluationTransform'].enum_type = _GLMREGRESSOR_POSTEVALUATIONTRANSFORM 133 | _GLMREGRESSOR_POSTEVALUATIONTRANSFORM.containing_type = _GLMREGRESSOR 134 | DESCRIPTOR.message_types_by_name['GLMRegressor'] = _GLMREGRESSOR 135 | 136 | GLMRegressor = _reflection.GeneratedProtocolMessageType('GLMRegressor', (_message.Message,), dict( 137 | 138 | DoubleArray = _reflection.GeneratedProtocolMessageType('DoubleArray', (_message.Message,), dict( 139 | DESCRIPTOR = _GLMREGRESSOR_DOUBLEARRAY, 140 | __module__ = 'GLMRegressor_pb2' 141 | # @@protoc_insertion_point(class_scope:CoreML.Specification.GLMRegressor.DoubleArray) 142 | )) 143 | , 144 | DESCRIPTOR = _GLMREGRESSOR, 145 | __module__ = 'GLMRegressor_pb2' 146 | # @@protoc_insertion_point(class_scope:CoreML.Specification.GLMRegressor) 147 | )) 148 | _sym_db.RegisterMessage(GLMRegressor) 149 | _sym_db.RegisterMessage(GLMRegressor.DoubleArray) 150 | 151 | 152 | DESCRIPTOR.has_options = True 153 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 154 | # @@protoc_insertion_point(module_scope) 155 | -------------------------------------------------------------------------------- /Identity_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: Identity.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='Identity.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x0eIdentity.proto\x12\x14\x43oreML.Specification\"\n\n\x08IdentityB\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | 29 | _IDENTITY = _descriptor.Descriptor( 30 | name='Identity', 31 | full_name='CoreML.Specification.Identity', 32 | filename=None, 33 | file=DESCRIPTOR, 34 | containing_type=None, 35 | fields=[ 36 | ], 37 | extensions=[ 38 | ], 39 | nested_types=[], 40 | enum_types=[ 41 | ], 42 | options=None, 43 | is_extendable=False, 44 | syntax='proto3', 45 | extension_ranges=[], 46 | oneofs=[ 47 | ], 48 | serialized_start=40, 49 | serialized_end=50, 50 | ) 51 | 52 | DESCRIPTOR.message_types_by_name['Identity'] = _IDENTITY 53 | 54 | Identity = _reflection.GeneratedProtocolMessageType('Identity', (_message.Message,), dict( 55 | DESCRIPTOR = _IDENTITY, 56 | __module__ = 'Identity_pb2' 57 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Identity) 58 | )) 59 | _sym_db.RegisterMessage(Identity) 60 | 61 | 62 | DESCRIPTOR.has_options = True 63 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 64 | # @@protoc_insertion_point(module_scope) 65 | -------------------------------------------------------------------------------- /Imputer_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: Imputer.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import DataStructures_pb2 as DataStructures__pb2 17 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 18 | 19 | from DataStructures_pb2 import * 20 | 21 | DESCRIPTOR = _descriptor.FileDescriptor( 22 | name='Imputer.proto', 23 | package='CoreML.Specification', 24 | syntax='proto3', 25 | serialized_pb=_b('\n\rImputer.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\xf3\x03\n\x07Imputer\x12\x1c\n\x12imputedDoubleValue\x18\x01 \x01(\x01H\x00\x12\x1b\n\x11imputedInt64Value\x18\x02 \x01(\x03H\x00\x12\x1c\n\x12imputedStringValue\x18\x03 \x01(\tH\x00\x12@\n\x12imputedDoubleArray\x18\x04 \x01(\x0b\x32\".CoreML.Specification.DoubleVectorH\x00\x12>\n\x11imputedInt64Array\x18\x05 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x00\x12J\n\x17imputedStringDictionary\x18\x06 \x01(\x0b\x32\'.CoreML.Specification.StringToDoubleMapH\x00\x12H\n\x16imputedInt64Dictionary\x18\x07 \x01(\x0b\x32&.CoreML.Specification.Int64ToDoubleMapH\x00\x12\x1c\n\x12replaceDoubleValue\x18\x0b \x01(\x01H\x01\x12\x1b\n\x11replaceInt64Value\x18\x0c \x01(\x03H\x01\x12\x1c\n\x12replaceStringValue\x18\r \x01(\tH\x01\x42\x0e\n\x0cImputedValueB\x0e\n\x0cReplaceValueB\x02H\x03P\x00\x62\x06proto3') 26 | , 27 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 28 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 29 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 30 | 31 | 32 | 33 | 34 | _IMPUTER = _descriptor.Descriptor( 35 | name='Imputer', 36 | full_name='CoreML.Specification.Imputer', 37 | filename=None, 38 | file=DESCRIPTOR, 39 | containing_type=None, 40 | fields=[ 41 | _descriptor.FieldDescriptor( 42 | name='imputedDoubleValue', full_name='CoreML.Specification.Imputer.imputedDoubleValue', index=0, 43 | number=1, type=1, cpp_type=5, label=1, 44 | has_default_value=False, default_value=float(0), 45 | message_type=None, enum_type=None, containing_type=None, 46 | is_extension=False, extension_scope=None, 47 | options=None), 48 | _descriptor.FieldDescriptor( 49 | name='imputedInt64Value', full_name='CoreML.Specification.Imputer.imputedInt64Value', index=1, 50 | number=2, type=3, cpp_type=2, label=1, 51 | has_default_value=False, default_value=0, 52 | message_type=None, enum_type=None, containing_type=None, 53 | is_extension=False, extension_scope=None, 54 | options=None), 55 | _descriptor.FieldDescriptor( 56 | name='imputedStringValue', full_name='CoreML.Specification.Imputer.imputedStringValue', index=2, 57 | number=3, type=9, cpp_type=9, label=1, 58 | has_default_value=False, default_value=_b("").decode('utf-8'), 59 | message_type=None, enum_type=None, containing_type=None, 60 | is_extension=False, extension_scope=None, 61 | options=None), 62 | _descriptor.FieldDescriptor( 63 | name='imputedDoubleArray', full_name='CoreML.Specification.Imputer.imputedDoubleArray', index=3, 64 | number=4, type=11, cpp_type=10, label=1, 65 | has_default_value=False, default_value=None, 66 | message_type=None, enum_type=None, containing_type=None, 67 | is_extension=False, extension_scope=None, 68 | options=None), 69 | _descriptor.FieldDescriptor( 70 | name='imputedInt64Array', full_name='CoreML.Specification.Imputer.imputedInt64Array', index=4, 71 | number=5, type=11, cpp_type=10, label=1, 72 | has_default_value=False, default_value=None, 73 | message_type=None, enum_type=None, containing_type=None, 74 | is_extension=False, extension_scope=None, 75 | options=None), 76 | _descriptor.FieldDescriptor( 77 | name='imputedStringDictionary', full_name='CoreML.Specification.Imputer.imputedStringDictionary', index=5, 78 | number=6, type=11, cpp_type=10, label=1, 79 | has_default_value=False, default_value=None, 80 | message_type=None, enum_type=None, containing_type=None, 81 | is_extension=False, extension_scope=None, 82 | options=None), 83 | _descriptor.FieldDescriptor( 84 | name='imputedInt64Dictionary', full_name='CoreML.Specification.Imputer.imputedInt64Dictionary', index=6, 85 | number=7, type=11, cpp_type=10, label=1, 86 | has_default_value=False, default_value=None, 87 | message_type=None, enum_type=None, containing_type=None, 88 | is_extension=False, extension_scope=None, 89 | options=None), 90 | _descriptor.FieldDescriptor( 91 | name='replaceDoubleValue', full_name='CoreML.Specification.Imputer.replaceDoubleValue', index=7, 92 | number=11, type=1, cpp_type=5, label=1, 93 | has_default_value=False, default_value=float(0), 94 | message_type=None, enum_type=None, containing_type=None, 95 | is_extension=False, extension_scope=None, 96 | options=None), 97 | _descriptor.FieldDescriptor( 98 | name='replaceInt64Value', full_name='CoreML.Specification.Imputer.replaceInt64Value', index=8, 99 | number=12, type=3, cpp_type=2, label=1, 100 | has_default_value=False, default_value=0, 101 | message_type=None, enum_type=None, containing_type=None, 102 | is_extension=False, extension_scope=None, 103 | options=None), 104 | _descriptor.FieldDescriptor( 105 | name='replaceStringValue', full_name='CoreML.Specification.Imputer.replaceStringValue', index=9, 106 | number=13, type=9, cpp_type=9, label=1, 107 | has_default_value=False, default_value=_b("").decode('utf-8'), 108 | message_type=None, enum_type=None, containing_type=None, 109 | is_extension=False, extension_scope=None, 110 | options=None), 111 | ], 112 | extensions=[ 113 | ], 114 | nested_types=[], 115 | enum_types=[ 116 | ], 117 | options=None, 118 | is_extendable=False, 119 | syntax='proto3', 120 | extension_ranges=[], 121 | oneofs=[ 122 | _descriptor.OneofDescriptor( 123 | name='ImputedValue', full_name='CoreML.Specification.Imputer.ImputedValue', 124 | index=0, containing_type=None, fields=[]), 125 | _descriptor.OneofDescriptor( 126 | name='ReplaceValue', full_name='CoreML.Specification.Imputer.ReplaceValue', 127 | index=1, containing_type=None, fields=[]), 128 | ], 129 | serialized_start=62, 130 | serialized_end=561, 131 | ) 132 | 133 | _IMPUTER.fields_by_name['imputedDoubleArray'].message_type = DataStructures__pb2._DOUBLEVECTOR 134 | _IMPUTER.fields_by_name['imputedInt64Array'].message_type = DataStructures__pb2._INT64VECTOR 135 | _IMPUTER.fields_by_name['imputedStringDictionary'].message_type = DataStructures__pb2._STRINGTODOUBLEMAP 136 | _IMPUTER.fields_by_name['imputedInt64Dictionary'].message_type = DataStructures__pb2._INT64TODOUBLEMAP 137 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 138 | _IMPUTER.fields_by_name['imputedDoubleValue']) 139 | _IMPUTER.fields_by_name['imputedDoubleValue'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 140 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 141 | _IMPUTER.fields_by_name['imputedInt64Value']) 142 | _IMPUTER.fields_by_name['imputedInt64Value'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 143 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 144 | _IMPUTER.fields_by_name['imputedStringValue']) 145 | _IMPUTER.fields_by_name['imputedStringValue'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 146 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 147 | _IMPUTER.fields_by_name['imputedDoubleArray']) 148 | _IMPUTER.fields_by_name['imputedDoubleArray'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 149 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 150 | _IMPUTER.fields_by_name['imputedInt64Array']) 151 | _IMPUTER.fields_by_name['imputedInt64Array'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 152 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 153 | _IMPUTER.fields_by_name['imputedStringDictionary']) 154 | _IMPUTER.fields_by_name['imputedStringDictionary'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 155 | _IMPUTER.oneofs_by_name['ImputedValue'].fields.append( 156 | _IMPUTER.fields_by_name['imputedInt64Dictionary']) 157 | _IMPUTER.fields_by_name['imputedInt64Dictionary'].containing_oneof = _IMPUTER.oneofs_by_name['ImputedValue'] 158 | _IMPUTER.oneofs_by_name['ReplaceValue'].fields.append( 159 | _IMPUTER.fields_by_name['replaceDoubleValue']) 160 | _IMPUTER.fields_by_name['replaceDoubleValue'].containing_oneof = _IMPUTER.oneofs_by_name['ReplaceValue'] 161 | _IMPUTER.oneofs_by_name['ReplaceValue'].fields.append( 162 | _IMPUTER.fields_by_name['replaceInt64Value']) 163 | _IMPUTER.fields_by_name['replaceInt64Value'].containing_oneof = _IMPUTER.oneofs_by_name['ReplaceValue'] 164 | _IMPUTER.oneofs_by_name['ReplaceValue'].fields.append( 165 | _IMPUTER.fields_by_name['replaceStringValue']) 166 | _IMPUTER.fields_by_name['replaceStringValue'].containing_oneof = _IMPUTER.oneofs_by_name['ReplaceValue'] 167 | DESCRIPTOR.message_types_by_name['Imputer'] = _IMPUTER 168 | 169 | Imputer = _reflection.GeneratedProtocolMessageType('Imputer', (_message.Message,), dict( 170 | DESCRIPTOR = _IMPUTER, 171 | __module__ = 'Imputer_pb2' 172 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Imputer) 173 | )) 174 | _sym_db.RegisterMessage(Imputer) 175 | 176 | 177 | DESCRIPTOR.has_options = True 178 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 179 | # @@protoc_insertion_point(module_scope) 180 | -------------------------------------------------------------------------------- /Model_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: Model.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import ArrayFeatureExtractor_pb2 as ArrayFeatureExtractor__pb2 17 | import CategoricalMapping_pb2 as CategoricalMapping__pb2 18 | DataStructures__pb2 = CategoricalMapping__pb2.DataStructures__pb2 19 | FeatureTypes__pb2 = CategoricalMapping__pb2.FeatureTypes__pb2 20 | import DictVectorizer_pb2 as DictVectorizer__pb2 21 | DataStructures__pb2 = DictVectorizer__pb2.DataStructures__pb2 22 | FeatureTypes__pb2 = DictVectorizer__pb2.FeatureTypes__pb2 23 | import FeatureTypes_pb2 as FeatureTypes__pb2 24 | import FeatureVectorizer_pb2 as FeatureVectorizer__pb2 25 | import GLMRegressor_pb2 as GLMRegressor__pb2 26 | import GLMClassifier_pb2 as GLMClassifier__pb2 27 | DataStructures__pb2 = GLMClassifier__pb2.DataStructures__pb2 28 | FeatureTypes__pb2 = GLMClassifier__pb2.FeatureTypes__pb2 29 | import Identity_pb2 as Identity__pb2 30 | import Imputer_pb2 as Imputer__pb2 31 | DataStructures__pb2 = Imputer__pb2.DataStructures__pb2 32 | FeatureTypes__pb2 = Imputer__pb2.FeatureTypes__pb2 33 | import NeuralNetwork_pb2 as NeuralNetwork__pb2 34 | DataStructures__pb2 = NeuralNetwork__pb2.DataStructures__pb2 35 | FeatureTypes__pb2 = NeuralNetwork__pb2.FeatureTypes__pb2 36 | import Normalizer_pb2 as Normalizer__pb2 37 | import OneHotEncoder_pb2 as OneHotEncoder__pb2 38 | DataStructures__pb2 = OneHotEncoder__pb2.DataStructures__pb2 39 | FeatureTypes__pb2 = OneHotEncoder__pb2.FeatureTypes__pb2 40 | import Scaler_pb2 as Scaler__pb2 41 | import SVM_pb2 as SVM__pb2 42 | DataStructures__pb2 = SVM__pb2.DataStructures__pb2 43 | FeatureTypes__pb2 = SVM__pb2.FeatureTypes__pb2 44 | import TreeEnsemble_pb2 as TreeEnsemble__pb2 45 | DataStructures__pb2 = TreeEnsemble__pb2.DataStructures__pb2 46 | FeatureTypes__pb2 = TreeEnsemble__pb2.FeatureTypes__pb2 47 | 48 | from ArrayFeatureExtractor_pb2 import * 49 | from CategoricalMapping_pb2 import * 50 | from DictVectorizer_pb2 import * 51 | from FeatureTypes_pb2 import * 52 | from FeatureVectorizer_pb2 import * 53 | from GLMRegressor_pb2 import * 54 | from GLMClassifier_pb2 import * 55 | from Identity_pb2 import * 56 | from Imputer_pb2 import * 57 | from NeuralNetwork_pb2 import * 58 | from Normalizer_pb2 import * 59 | from OneHotEncoder_pb2 import * 60 | from Scaler_pb2 import * 61 | from SVM_pb2 import * 62 | from TreeEnsemble_pb2 import * 63 | 64 | DESCRIPTOR = _descriptor.FileDescriptor( 65 | name='Model.proto', 66 | package='CoreML.Specification', 67 | syntax='proto3', 68 | serialized_pb=_b('\n\x0bModel.proto\x12\x14\x43oreML.Specification\x1a\x1b\x41rrayFeatureExtractor.proto\x1a\x18\x43\x61tegoricalMapping.proto\x1a\x14\x44ictVectorizer.proto\x1a\x12\x46\x65\x61tureTypes.proto\x1a\x17\x46\x65\x61tureVectorizer.proto\x1a\x12GLMRegressor.proto\x1a\x13GLMClassifier.proto\x1a\x0eIdentity.proto\x1a\rImputer.proto\x1a\x13NeuralNetwork.proto\x1a\x10Normalizer.proto\x1a\x13OneHotEncoder.proto\x1a\x0cScaler.proto\x1a\tSVM.proto\x1a\x12TreeEnsemble.proto\"7\n\x08Pipeline\x12+\n\x06models\x18\x01 \x03(\x0b\x32\x1b.CoreML.Specification.Model\"\xd5\x01\n\x12PipelineClassifier\x12\x30\n\x08pipeline\x18\x01 \x01(\x0b\x32\x1e.CoreML.Specification.Pipeline\x12?\n\x11stringClassLabels\x18\x64 \x01(\x0b\x32\".CoreML.Specification.StringVectorH\x00\x12=\n\x10int64ClassLabels\x18\x65 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x00\x42\r\n\x0b\x43lassLabels\"E\n\x11PipelineRegressor\x12\x30\n\x08pipeline\x18\x01 \x01(\x0b\x32\x1e.CoreML.Specification.Pipeline\"m\n\x12\x46\x65\x61tureDescription\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x18\n\x10shortDescription\x18\x02 \x01(\t\x12/\n\x04type\x18\x03 \x01(\x0b\x32!.CoreML.Specification.FeatureType\"\xd6\x01\n\x08Metadata\x12\x18\n\x10shortDescription\x18\x01 \x01(\t\x12\x15\n\rversionString\x18\x02 \x01(\t\x12\x0e\n\x06\x61uthor\x18\x03 \x01(\t\x12\x0f\n\x07license\x18\x04 \x01(\t\x12\x44\n\x0buserDefined\x18\x64 \x03(\x0b\x32/.CoreML.Specification.Metadata.UserDefinedEntry\x1a\x32\n\x10UserDefinedEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xf9\x01\n\x10ModelDescription\x12\x37\n\x05input\x18\x01 \x03(\x0b\x32(.CoreML.Specification.FeatureDescription\x12\x38\n\x06output\x18\n \x03(\x0b\x32(.CoreML.Specification.FeatureDescription\x12\x1c\n\x14predictedFeatureName\x18\x0b \x01(\t\x12\"\n\x1apredictedProbabilitiesName\x18\x0c \x01(\t\x12\x30\n\x08metadata\x18\x64 \x01(\x0b\x32\x1e.CoreML.Specification.Metadata\"\x83\x0c\n\x05Model\x12\x1c\n\x14specificationVersion\x18\x01 \x01(\x05\x12;\n\x0b\x64\x65scription\x18\x02 \x01(\x0b\x32&.CoreML.Specification.ModelDescription\x12G\n\x12pipelineClassifier\x18\xc8\x01 \x01(\x0b\x32(.CoreML.Specification.PipelineClassifierH\x00\x12\x45\n\x11pipelineRegressor\x18\xc9\x01 \x01(\x0b\x32\'.CoreML.Specification.PipelineRegressorH\x00\x12\x33\n\x08pipeline\x18\xca\x01 \x01(\x0b\x32\x1e.CoreML.Specification.PipelineH\x00\x12;\n\x0cglmRegressor\x18\xac\x02 \x01(\x0b\x32\".CoreML.Specification.GLMRegressorH\x00\x12O\n\x16supportVectorRegressor\x18\xad\x02 \x01(\x0b\x32,.CoreML.Specification.SupportVectorRegressorH\x00\x12M\n\x15treeEnsembleRegressor\x18\xae\x02 \x01(\x0b\x32+.CoreML.Specification.TreeEnsembleRegressorH\x00\x12O\n\x16neuralNetworkRegressor\x18\xaf\x02 \x01(\x0b\x32,.CoreML.Specification.NeuralNetworkRegressorH\x00\x12=\n\rglmClassifier\x18\x90\x03 \x01(\x0b\x32#.CoreML.Specification.GLMClassifierH\x00\x12Q\n\x17supportVectorClassifier\x18\x91\x03 \x01(\x0b\x32-.CoreML.Specification.SupportVectorClassifierH\x00\x12O\n\x16treeEnsembleClassifier\x18\x92\x03 \x01(\x0b\x32,.CoreML.Specification.TreeEnsembleClassifierH\x00\x12Q\n\x17neuralNetworkClassifier\x18\x93\x03 \x01(\x0b\x32-.CoreML.Specification.NeuralNetworkClassifierH\x00\x12=\n\rneuralNetwork\x18\xf4\x03 \x01(\x0b\x32#.CoreML.Specification.NeuralNetworkH\x00\x12=\n\roneHotEncoder\x18\xd8\x04 \x01(\x0b\x32#.CoreML.Specification.OneHotEncoderH\x00\x12\x31\n\x07imputer\x18\xd9\x04 \x01(\x0b\x32\x1d.CoreML.Specification.ImputerH\x00\x12\x45\n\x11\x66\x65\x61tureVectorizer\x18\xda\x04 \x01(\x0b\x32\'.CoreML.Specification.FeatureVectorizerH\x00\x12?\n\x0e\x64ictVectorizer\x18\xdb\x04 \x01(\x0b\x32$.CoreML.Specification.DictVectorizerH\x00\x12/\n\x06scaler\x18\xdc\x04 \x01(\x0b\x32\x1c.CoreML.Specification.ScalerH\x00\x12G\n\x12\x63\x61tegoricalMapping\x18\xde\x04 \x01(\x0b\x32(.CoreML.Specification.CategoricalMappingH\x00\x12\x37\n\nnormalizer\x18\xdf\x04 \x01(\x0b\x32 .CoreML.Specification.NormalizerH\x00\x12M\n\x15\x61rrayFeatureExtractor\x18\xe1\x04 \x01(\x0b\x32+.CoreML.Specification.ArrayFeatureExtractorH\x00\x12\x33\n\x08identity\x18\x84\x07 \x01(\x0b\x32\x1e.CoreML.Specification.IdentityH\x00\x42\x06\n\x04TypeB\x02H\x03P\x00P\x01P\x02P\x03P\x04P\x05P\x06P\x07P\x08P\tP\nP\x0bP\x0cP\rP\x0e\x62\x06proto3') 69 | , 70 | dependencies=[ArrayFeatureExtractor__pb2.DESCRIPTOR,CategoricalMapping__pb2.DESCRIPTOR,DictVectorizer__pb2.DESCRIPTOR,FeatureTypes__pb2.DESCRIPTOR,FeatureVectorizer__pb2.DESCRIPTOR,GLMRegressor__pb2.DESCRIPTOR,GLMClassifier__pb2.DESCRIPTOR,Identity__pb2.DESCRIPTOR,Imputer__pb2.DESCRIPTOR,NeuralNetwork__pb2.DESCRIPTOR,Normalizer__pb2.DESCRIPTOR,OneHotEncoder__pb2.DESCRIPTOR,Scaler__pb2.DESCRIPTOR,SVM__pb2.DESCRIPTOR,TreeEnsemble__pb2.DESCRIPTOR,], 71 | public_dependencies=[ArrayFeatureExtractor__pb2.DESCRIPTOR,CategoricalMapping__pb2.DESCRIPTOR,DictVectorizer__pb2.DESCRIPTOR,FeatureTypes__pb2.DESCRIPTOR,FeatureVectorizer__pb2.DESCRIPTOR,GLMRegressor__pb2.DESCRIPTOR,GLMClassifier__pb2.DESCRIPTOR,Identity__pb2.DESCRIPTOR,Imputer__pb2.DESCRIPTOR,NeuralNetwork__pb2.DESCRIPTOR,Normalizer__pb2.DESCRIPTOR,OneHotEncoder__pb2.DESCRIPTOR,Scaler__pb2.DESCRIPTOR,SVM__pb2.DESCRIPTOR,TreeEnsemble__pb2.DESCRIPTOR,]) 72 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 73 | 74 | 75 | 76 | 77 | _PIPELINE = _descriptor.Descriptor( 78 | name='Pipeline', 79 | full_name='CoreML.Specification.Pipeline', 80 | filename=None, 81 | file=DESCRIPTOR, 82 | containing_type=None, 83 | fields=[ 84 | _descriptor.FieldDescriptor( 85 | name='models', full_name='CoreML.Specification.Pipeline.models', index=0, 86 | number=1, type=11, cpp_type=10, label=3, 87 | has_default_value=False, default_value=[], 88 | message_type=None, enum_type=None, containing_type=None, 89 | is_extension=False, extension_scope=None, 90 | options=None), 91 | ], 92 | extensions=[ 93 | ], 94 | nested_types=[], 95 | enum_types=[ 96 | ], 97 | options=None, 98 | is_extendable=False, 99 | syntax='proto3', 100 | extension_ranges=[], 101 | oneofs=[ 102 | ], 103 | serialized_start=336, 104 | serialized_end=391, 105 | ) 106 | 107 | 108 | _PIPELINECLASSIFIER = _descriptor.Descriptor( 109 | name='PipelineClassifier', 110 | full_name='CoreML.Specification.PipelineClassifier', 111 | filename=None, 112 | file=DESCRIPTOR, 113 | containing_type=None, 114 | fields=[ 115 | _descriptor.FieldDescriptor( 116 | name='pipeline', full_name='CoreML.Specification.PipelineClassifier.pipeline', index=0, 117 | number=1, type=11, cpp_type=10, label=1, 118 | has_default_value=False, default_value=None, 119 | message_type=None, enum_type=None, containing_type=None, 120 | is_extension=False, extension_scope=None, 121 | options=None), 122 | _descriptor.FieldDescriptor( 123 | name='stringClassLabels', full_name='CoreML.Specification.PipelineClassifier.stringClassLabels', index=1, 124 | number=100, type=11, cpp_type=10, label=1, 125 | has_default_value=False, default_value=None, 126 | message_type=None, enum_type=None, containing_type=None, 127 | is_extension=False, extension_scope=None, 128 | options=None), 129 | _descriptor.FieldDescriptor( 130 | name='int64ClassLabels', full_name='CoreML.Specification.PipelineClassifier.int64ClassLabels', index=2, 131 | number=101, type=11, cpp_type=10, label=1, 132 | has_default_value=False, default_value=None, 133 | message_type=None, enum_type=None, containing_type=None, 134 | is_extension=False, extension_scope=None, 135 | options=None), 136 | ], 137 | extensions=[ 138 | ], 139 | nested_types=[], 140 | enum_types=[ 141 | ], 142 | options=None, 143 | is_extendable=False, 144 | syntax='proto3', 145 | extension_ranges=[], 146 | oneofs=[ 147 | _descriptor.OneofDescriptor( 148 | name='ClassLabels', full_name='CoreML.Specification.PipelineClassifier.ClassLabels', 149 | index=0, containing_type=None, fields=[]), 150 | ], 151 | serialized_start=394, 152 | serialized_end=607, 153 | ) 154 | 155 | 156 | _PIPELINEREGRESSOR = _descriptor.Descriptor( 157 | name='PipelineRegressor', 158 | full_name='CoreML.Specification.PipelineRegressor', 159 | filename=None, 160 | file=DESCRIPTOR, 161 | containing_type=None, 162 | fields=[ 163 | _descriptor.FieldDescriptor( 164 | name='pipeline', full_name='CoreML.Specification.PipelineRegressor.pipeline', index=0, 165 | number=1, type=11, cpp_type=10, label=1, 166 | has_default_value=False, default_value=None, 167 | message_type=None, enum_type=None, containing_type=None, 168 | is_extension=False, extension_scope=None, 169 | options=None), 170 | ], 171 | extensions=[ 172 | ], 173 | nested_types=[], 174 | enum_types=[ 175 | ], 176 | options=None, 177 | is_extendable=False, 178 | syntax='proto3', 179 | extension_ranges=[], 180 | oneofs=[ 181 | ], 182 | serialized_start=609, 183 | serialized_end=678, 184 | ) 185 | 186 | 187 | _FEATUREDESCRIPTION = _descriptor.Descriptor( 188 | name='FeatureDescription', 189 | full_name='CoreML.Specification.FeatureDescription', 190 | filename=None, 191 | file=DESCRIPTOR, 192 | containing_type=None, 193 | fields=[ 194 | _descriptor.FieldDescriptor( 195 | name='name', full_name='CoreML.Specification.FeatureDescription.name', index=0, 196 | number=1, type=9, cpp_type=9, label=1, 197 | has_default_value=False, default_value=_b("").decode('utf-8'), 198 | message_type=None, enum_type=None, containing_type=None, 199 | is_extension=False, extension_scope=None, 200 | options=None), 201 | _descriptor.FieldDescriptor( 202 | name='shortDescription', full_name='CoreML.Specification.FeatureDescription.shortDescription', index=1, 203 | number=2, type=9, cpp_type=9, label=1, 204 | has_default_value=False, default_value=_b("").decode('utf-8'), 205 | message_type=None, enum_type=None, containing_type=None, 206 | is_extension=False, extension_scope=None, 207 | options=None), 208 | _descriptor.FieldDescriptor( 209 | name='type', full_name='CoreML.Specification.FeatureDescription.type', index=2, 210 | number=3, type=11, cpp_type=10, label=1, 211 | has_default_value=False, default_value=None, 212 | message_type=None, enum_type=None, containing_type=None, 213 | is_extension=False, extension_scope=None, 214 | options=None), 215 | ], 216 | extensions=[ 217 | ], 218 | nested_types=[], 219 | enum_types=[ 220 | ], 221 | options=None, 222 | is_extendable=False, 223 | syntax='proto3', 224 | extension_ranges=[], 225 | oneofs=[ 226 | ], 227 | serialized_start=680, 228 | serialized_end=789, 229 | ) 230 | 231 | 232 | _METADATA_USERDEFINEDENTRY = _descriptor.Descriptor( 233 | name='UserDefinedEntry', 234 | full_name='CoreML.Specification.Metadata.UserDefinedEntry', 235 | filename=None, 236 | file=DESCRIPTOR, 237 | containing_type=None, 238 | fields=[ 239 | _descriptor.FieldDescriptor( 240 | name='key', full_name='CoreML.Specification.Metadata.UserDefinedEntry.key', index=0, 241 | number=1, type=9, cpp_type=9, label=1, 242 | has_default_value=False, default_value=_b("").decode('utf-8'), 243 | message_type=None, enum_type=None, containing_type=None, 244 | is_extension=False, extension_scope=None, 245 | options=None), 246 | _descriptor.FieldDescriptor( 247 | name='value', full_name='CoreML.Specification.Metadata.UserDefinedEntry.value', index=1, 248 | number=2, type=9, cpp_type=9, label=1, 249 | has_default_value=False, default_value=_b("").decode('utf-8'), 250 | message_type=None, enum_type=None, containing_type=None, 251 | is_extension=False, extension_scope=None, 252 | options=None), 253 | ], 254 | extensions=[ 255 | ], 256 | nested_types=[], 257 | enum_types=[ 258 | ], 259 | options=_descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')), 260 | is_extendable=False, 261 | syntax='proto3', 262 | extension_ranges=[], 263 | oneofs=[ 264 | ], 265 | serialized_start=956, 266 | serialized_end=1006, 267 | ) 268 | 269 | _METADATA = _descriptor.Descriptor( 270 | name='Metadata', 271 | full_name='CoreML.Specification.Metadata', 272 | filename=None, 273 | file=DESCRIPTOR, 274 | containing_type=None, 275 | fields=[ 276 | _descriptor.FieldDescriptor( 277 | name='shortDescription', full_name='CoreML.Specification.Metadata.shortDescription', index=0, 278 | number=1, type=9, cpp_type=9, label=1, 279 | has_default_value=False, default_value=_b("").decode('utf-8'), 280 | message_type=None, enum_type=None, containing_type=None, 281 | is_extension=False, extension_scope=None, 282 | options=None), 283 | _descriptor.FieldDescriptor( 284 | name='versionString', full_name='CoreML.Specification.Metadata.versionString', index=1, 285 | number=2, type=9, cpp_type=9, label=1, 286 | has_default_value=False, default_value=_b("").decode('utf-8'), 287 | message_type=None, enum_type=None, containing_type=None, 288 | is_extension=False, extension_scope=None, 289 | options=None), 290 | _descriptor.FieldDescriptor( 291 | name='author', full_name='CoreML.Specification.Metadata.author', index=2, 292 | number=3, type=9, cpp_type=9, label=1, 293 | has_default_value=False, default_value=_b("").decode('utf-8'), 294 | message_type=None, enum_type=None, containing_type=None, 295 | is_extension=False, extension_scope=None, 296 | options=None), 297 | _descriptor.FieldDescriptor( 298 | name='license', full_name='CoreML.Specification.Metadata.license', index=3, 299 | number=4, type=9, cpp_type=9, label=1, 300 | has_default_value=False, default_value=_b("").decode('utf-8'), 301 | message_type=None, enum_type=None, containing_type=None, 302 | is_extension=False, extension_scope=None, 303 | options=None), 304 | _descriptor.FieldDescriptor( 305 | name='userDefined', full_name='CoreML.Specification.Metadata.userDefined', index=4, 306 | number=100, type=11, cpp_type=10, label=3, 307 | has_default_value=False, default_value=[], 308 | message_type=None, enum_type=None, containing_type=None, 309 | is_extension=False, extension_scope=None, 310 | options=None), 311 | ], 312 | extensions=[ 313 | ], 314 | nested_types=[_METADATA_USERDEFINEDENTRY, ], 315 | enum_types=[ 316 | ], 317 | options=None, 318 | is_extendable=False, 319 | syntax='proto3', 320 | extension_ranges=[], 321 | oneofs=[ 322 | ], 323 | serialized_start=792, 324 | serialized_end=1006, 325 | ) 326 | 327 | 328 | _MODELDESCRIPTION = _descriptor.Descriptor( 329 | name='ModelDescription', 330 | full_name='CoreML.Specification.ModelDescription', 331 | filename=None, 332 | file=DESCRIPTOR, 333 | containing_type=None, 334 | fields=[ 335 | _descriptor.FieldDescriptor( 336 | name='input', full_name='CoreML.Specification.ModelDescription.input', index=0, 337 | number=1, type=11, cpp_type=10, label=3, 338 | has_default_value=False, default_value=[], 339 | message_type=None, enum_type=None, containing_type=None, 340 | is_extension=False, extension_scope=None, 341 | options=None), 342 | _descriptor.FieldDescriptor( 343 | name='output', full_name='CoreML.Specification.ModelDescription.output', index=1, 344 | number=10, type=11, cpp_type=10, label=3, 345 | has_default_value=False, default_value=[], 346 | message_type=None, enum_type=None, containing_type=None, 347 | is_extension=False, extension_scope=None, 348 | options=None), 349 | _descriptor.FieldDescriptor( 350 | name='predictedFeatureName', full_name='CoreML.Specification.ModelDescription.predictedFeatureName', index=2, 351 | number=11, type=9, cpp_type=9, label=1, 352 | has_default_value=False, default_value=_b("").decode('utf-8'), 353 | message_type=None, enum_type=None, containing_type=None, 354 | is_extension=False, extension_scope=None, 355 | options=None), 356 | _descriptor.FieldDescriptor( 357 | name='predictedProbabilitiesName', full_name='CoreML.Specification.ModelDescription.predictedProbabilitiesName', index=3, 358 | number=12, type=9, cpp_type=9, label=1, 359 | has_default_value=False, default_value=_b("").decode('utf-8'), 360 | message_type=None, enum_type=None, containing_type=None, 361 | is_extension=False, extension_scope=None, 362 | options=None), 363 | _descriptor.FieldDescriptor( 364 | name='metadata', full_name='CoreML.Specification.ModelDescription.metadata', index=4, 365 | number=100, type=11, cpp_type=10, label=1, 366 | has_default_value=False, default_value=None, 367 | message_type=None, enum_type=None, containing_type=None, 368 | is_extension=False, extension_scope=None, 369 | options=None), 370 | ], 371 | extensions=[ 372 | ], 373 | nested_types=[], 374 | enum_types=[ 375 | ], 376 | options=None, 377 | is_extendable=False, 378 | syntax='proto3', 379 | extension_ranges=[], 380 | oneofs=[ 381 | ], 382 | serialized_start=1009, 383 | serialized_end=1258, 384 | ) 385 | 386 | 387 | _MODEL = _descriptor.Descriptor( 388 | name='Model', 389 | full_name='CoreML.Specification.Model', 390 | filename=None, 391 | file=DESCRIPTOR, 392 | containing_type=None, 393 | fields=[ 394 | _descriptor.FieldDescriptor( 395 | name='specificationVersion', full_name='CoreML.Specification.Model.specificationVersion', index=0, 396 | number=1, type=5, cpp_type=1, label=1, 397 | has_default_value=False, default_value=0, 398 | message_type=None, enum_type=None, containing_type=None, 399 | is_extension=False, extension_scope=None, 400 | options=None), 401 | _descriptor.FieldDescriptor( 402 | name='description', full_name='CoreML.Specification.Model.description', index=1, 403 | number=2, type=11, cpp_type=10, label=1, 404 | has_default_value=False, default_value=None, 405 | message_type=None, enum_type=None, containing_type=None, 406 | is_extension=False, extension_scope=None, 407 | options=None), 408 | _descriptor.FieldDescriptor( 409 | name='pipelineClassifier', full_name='CoreML.Specification.Model.pipelineClassifier', index=2, 410 | number=200, type=11, cpp_type=10, label=1, 411 | has_default_value=False, default_value=None, 412 | message_type=None, enum_type=None, containing_type=None, 413 | is_extension=False, extension_scope=None, 414 | options=None), 415 | _descriptor.FieldDescriptor( 416 | name='pipelineRegressor', full_name='CoreML.Specification.Model.pipelineRegressor', index=3, 417 | number=201, type=11, cpp_type=10, label=1, 418 | has_default_value=False, default_value=None, 419 | message_type=None, enum_type=None, containing_type=None, 420 | is_extension=False, extension_scope=None, 421 | options=None), 422 | _descriptor.FieldDescriptor( 423 | name='pipeline', full_name='CoreML.Specification.Model.pipeline', index=4, 424 | number=202, type=11, cpp_type=10, label=1, 425 | has_default_value=False, default_value=None, 426 | message_type=None, enum_type=None, containing_type=None, 427 | is_extension=False, extension_scope=None, 428 | options=None), 429 | _descriptor.FieldDescriptor( 430 | name='glmRegressor', full_name='CoreML.Specification.Model.glmRegressor', index=5, 431 | number=300, type=11, cpp_type=10, label=1, 432 | has_default_value=False, default_value=None, 433 | message_type=None, enum_type=None, containing_type=None, 434 | is_extension=False, extension_scope=None, 435 | options=None), 436 | _descriptor.FieldDescriptor( 437 | name='supportVectorRegressor', full_name='CoreML.Specification.Model.supportVectorRegressor', index=6, 438 | number=301, type=11, cpp_type=10, label=1, 439 | has_default_value=False, default_value=None, 440 | message_type=None, enum_type=None, containing_type=None, 441 | is_extension=False, extension_scope=None, 442 | options=None), 443 | _descriptor.FieldDescriptor( 444 | name='treeEnsembleRegressor', full_name='CoreML.Specification.Model.treeEnsembleRegressor', index=7, 445 | number=302, type=11, cpp_type=10, label=1, 446 | has_default_value=False, default_value=None, 447 | message_type=None, enum_type=None, containing_type=None, 448 | is_extension=False, extension_scope=None, 449 | options=None), 450 | _descriptor.FieldDescriptor( 451 | name='neuralNetworkRegressor', full_name='CoreML.Specification.Model.neuralNetworkRegressor', index=8, 452 | number=303, type=11, cpp_type=10, label=1, 453 | has_default_value=False, default_value=None, 454 | message_type=None, enum_type=None, containing_type=None, 455 | is_extension=False, extension_scope=None, 456 | options=None), 457 | _descriptor.FieldDescriptor( 458 | name='glmClassifier', full_name='CoreML.Specification.Model.glmClassifier', index=9, 459 | number=400, type=11, cpp_type=10, label=1, 460 | has_default_value=False, default_value=None, 461 | message_type=None, enum_type=None, containing_type=None, 462 | is_extension=False, extension_scope=None, 463 | options=None), 464 | _descriptor.FieldDescriptor( 465 | name='supportVectorClassifier', full_name='CoreML.Specification.Model.supportVectorClassifier', index=10, 466 | number=401, type=11, cpp_type=10, label=1, 467 | has_default_value=False, default_value=None, 468 | message_type=None, enum_type=None, containing_type=None, 469 | is_extension=False, extension_scope=None, 470 | options=None), 471 | _descriptor.FieldDescriptor( 472 | name='treeEnsembleClassifier', full_name='CoreML.Specification.Model.treeEnsembleClassifier', index=11, 473 | number=402, type=11, cpp_type=10, label=1, 474 | has_default_value=False, default_value=None, 475 | message_type=None, enum_type=None, containing_type=None, 476 | is_extension=False, extension_scope=None, 477 | options=None), 478 | _descriptor.FieldDescriptor( 479 | name='neuralNetworkClassifier', full_name='CoreML.Specification.Model.neuralNetworkClassifier', index=12, 480 | number=403, type=11, cpp_type=10, label=1, 481 | has_default_value=False, default_value=None, 482 | message_type=None, enum_type=None, containing_type=None, 483 | is_extension=False, extension_scope=None, 484 | options=None), 485 | _descriptor.FieldDescriptor( 486 | name='neuralNetwork', full_name='CoreML.Specification.Model.neuralNetwork', index=13, 487 | number=500, type=11, cpp_type=10, label=1, 488 | has_default_value=False, default_value=None, 489 | message_type=None, enum_type=None, containing_type=None, 490 | is_extension=False, extension_scope=None, 491 | options=None), 492 | _descriptor.FieldDescriptor( 493 | name='oneHotEncoder', full_name='CoreML.Specification.Model.oneHotEncoder', index=14, 494 | number=600, type=11, cpp_type=10, label=1, 495 | has_default_value=False, default_value=None, 496 | message_type=None, enum_type=None, containing_type=None, 497 | is_extension=False, extension_scope=None, 498 | options=None), 499 | _descriptor.FieldDescriptor( 500 | name='imputer', full_name='CoreML.Specification.Model.imputer', index=15, 501 | number=601, type=11, cpp_type=10, label=1, 502 | has_default_value=False, default_value=None, 503 | message_type=None, enum_type=None, containing_type=None, 504 | is_extension=False, extension_scope=None, 505 | options=None), 506 | _descriptor.FieldDescriptor( 507 | name='featureVectorizer', full_name='CoreML.Specification.Model.featureVectorizer', index=16, 508 | number=602, type=11, cpp_type=10, label=1, 509 | has_default_value=False, default_value=None, 510 | message_type=None, enum_type=None, containing_type=None, 511 | is_extension=False, extension_scope=None, 512 | options=None), 513 | _descriptor.FieldDescriptor( 514 | name='dictVectorizer', full_name='CoreML.Specification.Model.dictVectorizer', index=17, 515 | number=603, type=11, cpp_type=10, label=1, 516 | has_default_value=False, default_value=None, 517 | message_type=None, enum_type=None, containing_type=None, 518 | is_extension=False, extension_scope=None, 519 | options=None), 520 | _descriptor.FieldDescriptor( 521 | name='scaler', full_name='CoreML.Specification.Model.scaler', index=18, 522 | number=604, type=11, cpp_type=10, label=1, 523 | has_default_value=False, default_value=None, 524 | message_type=None, enum_type=None, containing_type=None, 525 | is_extension=False, extension_scope=None, 526 | options=None), 527 | _descriptor.FieldDescriptor( 528 | name='categoricalMapping', full_name='CoreML.Specification.Model.categoricalMapping', index=19, 529 | number=606, type=11, cpp_type=10, label=1, 530 | has_default_value=False, default_value=None, 531 | message_type=None, enum_type=None, containing_type=None, 532 | is_extension=False, extension_scope=None, 533 | options=None), 534 | _descriptor.FieldDescriptor( 535 | name='normalizer', full_name='CoreML.Specification.Model.normalizer', index=20, 536 | number=607, type=11, cpp_type=10, label=1, 537 | has_default_value=False, default_value=None, 538 | message_type=None, enum_type=None, containing_type=None, 539 | is_extension=False, extension_scope=None, 540 | options=None), 541 | _descriptor.FieldDescriptor( 542 | name='arrayFeatureExtractor', full_name='CoreML.Specification.Model.arrayFeatureExtractor', index=21, 543 | number=609, type=11, cpp_type=10, label=1, 544 | has_default_value=False, default_value=None, 545 | message_type=None, enum_type=None, containing_type=None, 546 | is_extension=False, extension_scope=None, 547 | options=None), 548 | _descriptor.FieldDescriptor( 549 | name='identity', full_name='CoreML.Specification.Model.identity', index=22, 550 | number=900, type=11, cpp_type=10, label=1, 551 | has_default_value=False, default_value=None, 552 | message_type=None, enum_type=None, containing_type=None, 553 | is_extension=False, extension_scope=None, 554 | options=None), 555 | ], 556 | extensions=[ 557 | ], 558 | nested_types=[], 559 | enum_types=[ 560 | ], 561 | options=None, 562 | is_extendable=False, 563 | syntax='proto3', 564 | extension_ranges=[], 565 | oneofs=[ 566 | _descriptor.OneofDescriptor( 567 | name='Type', full_name='CoreML.Specification.Model.Type', 568 | index=0, containing_type=None, fields=[]), 569 | ], 570 | serialized_start=1261, 571 | serialized_end=2800, 572 | ) 573 | 574 | _PIPELINE.fields_by_name['models'].message_type = _MODEL 575 | _PIPELINECLASSIFIER.fields_by_name['pipeline'].message_type = _PIPELINE 576 | _PIPELINECLASSIFIER.fields_by_name['stringClassLabels'].message_type = DataStructures__pb2._STRINGVECTOR 577 | _PIPELINECLASSIFIER.fields_by_name['int64ClassLabels'].message_type = DataStructures__pb2._INT64VECTOR 578 | _PIPELINECLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 579 | _PIPELINECLASSIFIER.fields_by_name['stringClassLabels']) 580 | _PIPELINECLASSIFIER.fields_by_name['stringClassLabels'].containing_oneof = _PIPELINECLASSIFIER.oneofs_by_name['ClassLabels'] 581 | _PIPELINECLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 582 | _PIPELINECLASSIFIER.fields_by_name['int64ClassLabels']) 583 | _PIPELINECLASSIFIER.fields_by_name['int64ClassLabels'].containing_oneof = _PIPELINECLASSIFIER.oneofs_by_name['ClassLabels'] 584 | _PIPELINEREGRESSOR.fields_by_name['pipeline'].message_type = _PIPELINE 585 | _FEATUREDESCRIPTION.fields_by_name['type'].message_type = FeatureTypes__pb2._FEATURETYPE 586 | _METADATA_USERDEFINEDENTRY.containing_type = _METADATA 587 | _METADATA.fields_by_name['userDefined'].message_type = _METADATA_USERDEFINEDENTRY 588 | _MODELDESCRIPTION.fields_by_name['input'].message_type = _FEATUREDESCRIPTION 589 | _MODELDESCRIPTION.fields_by_name['output'].message_type = _FEATUREDESCRIPTION 590 | _MODELDESCRIPTION.fields_by_name['metadata'].message_type = _METADATA 591 | _MODEL.fields_by_name['description'].message_type = _MODELDESCRIPTION 592 | _MODEL.fields_by_name['pipelineClassifier'].message_type = _PIPELINECLASSIFIER 593 | _MODEL.fields_by_name['pipelineRegressor'].message_type = _PIPELINEREGRESSOR 594 | _MODEL.fields_by_name['pipeline'].message_type = _PIPELINE 595 | _MODEL.fields_by_name['glmRegressor'].message_type = GLMRegressor__pb2._GLMREGRESSOR 596 | _MODEL.fields_by_name['supportVectorRegressor'].message_type = SVM__pb2._SUPPORTVECTORREGRESSOR 597 | _MODEL.fields_by_name['treeEnsembleRegressor'].message_type = TreeEnsemble__pb2._TREEENSEMBLEREGRESSOR 598 | _MODEL.fields_by_name['neuralNetworkRegressor'].message_type = NeuralNetwork__pb2._NEURALNETWORKREGRESSOR 599 | _MODEL.fields_by_name['glmClassifier'].message_type = GLMClassifier__pb2._GLMCLASSIFIER 600 | _MODEL.fields_by_name['supportVectorClassifier'].message_type = SVM__pb2._SUPPORTVECTORCLASSIFIER 601 | _MODEL.fields_by_name['treeEnsembleClassifier'].message_type = TreeEnsemble__pb2._TREEENSEMBLECLASSIFIER 602 | _MODEL.fields_by_name['neuralNetworkClassifier'].message_type = NeuralNetwork__pb2._NEURALNETWORKCLASSIFIER 603 | _MODEL.fields_by_name['neuralNetwork'].message_type = NeuralNetwork__pb2._NEURALNETWORK 604 | _MODEL.fields_by_name['oneHotEncoder'].message_type = OneHotEncoder__pb2._ONEHOTENCODER 605 | _MODEL.fields_by_name['imputer'].message_type = Imputer__pb2._IMPUTER 606 | _MODEL.fields_by_name['featureVectorizer'].message_type = FeatureVectorizer__pb2._FEATUREVECTORIZER 607 | _MODEL.fields_by_name['dictVectorizer'].message_type = DictVectorizer__pb2._DICTVECTORIZER 608 | _MODEL.fields_by_name['scaler'].message_type = Scaler__pb2._SCALER 609 | _MODEL.fields_by_name['categoricalMapping'].message_type = CategoricalMapping__pb2._CATEGORICALMAPPING 610 | _MODEL.fields_by_name['normalizer'].message_type = Normalizer__pb2._NORMALIZER 611 | _MODEL.fields_by_name['arrayFeatureExtractor'].message_type = ArrayFeatureExtractor__pb2._ARRAYFEATUREEXTRACTOR 612 | _MODEL.fields_by_name['identity'].message_type = Identity__pb2._IDENTITY 613 | _MODEL.oneofs_by_name['Type'].fields.append( 614 | _MODEL.fields_by_name['pipelineClassifier']) 615 | _MODEL.fields_by_name['pipelineClassifier'].containing_oneof = _MODEL.oneofs_by_name['Type'] 616 | _MODEL.oneofs_by_name['Type'].fields.append( 617 | _MODEL.fields_by_name['pipelineRegressor']) 618 | _MODEL.fields_by_name['pipelineRegressor'].containing_oneof = _MODEL.oneofs_by_name['Type'] 619 | _MODEL.oneofs_by_name['Type'].fields.append( 620 | _MODEL.fields_by_name['pipeline']) 621 | _MODEL.fields_by_name['pipeline'].containing_oneof = _MODEL.oneofs_by_name['Type'] 622 | _MODEL.oneofs_by_name['Type'].fields.append( 623 | _MODEL.fields_by_name['glmRegressor']) 624 | _MODEL.fields_by_name['glmRegressor'].containing_oneof = _MODEL.oneofs_by_name['Type'] 625 | _MODEL.oneofs_by_name['Type'].fields.append( 626 | _MODEL.fields_by_name['supportVectorRegressor']) 627 | _MODEL.fields_by_name['supportVectorRegressor'].containing_oneof = _MODEL.oneofs_by_name['Type'] 628 | _MODEL.oneofs_by_name['Type'].fields.append( 629 | _MODEL.fields_by_name['treeEnsembleRegressor']) 630 | _MODEL.fields_by_name['treeEnsembleRegressor'].containing_oneof = _MODEL.oneofs_by_name['Type'] 631 | _MODEL.oneofs_by_name['Type'].fields.append( 632 | _MODEL.fields_by_name['neuralNetworkRegressor']) 633 | _MODEL.fields_by_name['neuralNetworkRegressor'].containing_oneof = _MODEL.oneofs_by_name['Type'] 634 | _MODEL.oneofs_by_name['Type'].fields.append( 635 | _MODEL.fields_by_name['glmClassifier']) 636 | _MODEL.fields_by_name['glmClassifier'].containing_oneof = _MODEL.oneofs_by_name['Type'] 637 | _MODEL.oneofs_by_name['Type'].fields.append( 638 | _MODEL.fields_by_name['supportVectorClassifier']) 639 | _MODEL.fields_by_name['supportVectorClassifier'].containing_oneof = _MODEL.oneofs_by_name['Type'] 640 | _MODEL.oneofs_by_name['Type'].fields.append( 641 | _MODEL.fields_by_name['treeEnsembleClassifier']) 642 | _MODEL.fields_by_name['treeEnsembleClassifier'].containing_oneof = _MODEL.oneofs_by_name['Type'] 643 | _MODEL.oneofs_by_name['Type'].fields.append( 644 | _MODEL.fields_by_name['neuralNetworkClassifier']) 645 | _MODEL.fields_by_name['neuralNetworkClassifier'].containing_oneof = _MODEL.oneofs_by_name['Type'] 646 | _MODEL.oneofs_by_name['Type'].fields.append( 647 | _MODEL.fields_by_name['neuralNetwork']) 648 | _MODEL.fields_by_name['neuralNetwork'].containing_oneof = _MODEL.oneofs_by_name['Type'] 649 | _MODEL.oneofs_by_name['Type'].fields.append( 650 | _MODEL.fields_by_name['oneHotEncoder']) 651 | _MODEL.fields_by_name['oneHotEncoder'].containing_oneof = _MODEL.oneofs_by_name['Type'] 652 | _MODEL.oneofs_by_name['Type'].fields.append( 653 | _MODEL.fields_by_name['imputer']) 654 | _MODEL.fields_by_name['imputer'].containing_oneof = _MODEL.oneofs_by_name['Type'] 655 | _MODEL.oneofs_by_name['Type'].fields.append( 656 | _MODEL.fields_by_name['featureVectorizer']) 657 | _MODEL.fields_by_name['featureVectorizer'].containing_oneof = _MODEL.oneofs_by_name['Type'] 658 | _MODEL.oneofs_by_name['Type'].fields.append( 659 | _MODEL.fields_by_name['dictVectorizer']) 660 | _MODEL.fields_by_name['dictVectorizer'].containing_oneof = _MODEL.oneofs_by_name['Type'] 661 | _MODEL.oneofs_by_name['Type'].fields.append( 662 | _MODEL.fields_by_name['scaler']) 663 | _MODEL.fields_by_name['scaler'].containing_oneof = _MODEL.oneofs_by_name['Type'] 664 | _MODEL.oneofs_by_name['Type'].fields.append( 665 | _MODEL.fields_by_name['categoricalMapping']) 666 | _MODEL.fields_by_name['categoricalMapping'].containing_oneof = _MODEL.oneofs_by_name['Type'] 667 | _MODEL.oneofs_by_name['Type'].fields.append( 668 | _MODEL.fields_by_name['normalizer']) 669 | _MODEL.fields_by_name['normalizer'].containing_oneof = _MODEL.oneofs_by_name['Type'] 670 | _MODEL.oneofs_by_name['Type'].fields.append( 671 | _MODEL.fields_by_name['arrayFeatureExtractor']) 672 | _MODEL.fields_by_name['arrayFeatureExtractor'].containing_oneof = _MODEL.oneofs_by_name['Type'] 673 | _MODEL.oneofs_by_name['Type'].fields.append( 674 | _MODEL.fields_by_name['identity']) 675 | _MODEL.fields_by_name['identity'].containing_oneof = _MODEL.oneofs_by_name['Type'] 676 | DESCRIPTOR.message_types_by_name['Pipeline'] = _PIPELINE 677 | DESCRIPTOR.message_types_by_name['PipelineClassifier'] = _PIPELINECLASSIFIER 678 | DESCRIPTOR.message_types_by_name['PipelineRegressor'] = _PIPELINEREGRESSOR 679 | DESCRIPTOR.message_types_by_name['FeatureDescription'] = _FEATUREDESCRIPTION 680 | DESCRIPTOR.message_types_by_name['Metadata'] = _METADATA 681 | DESCRIPTOR.message_types_by_name['ModelDescription'] = _MODELDESCRIPTION 682 | DESCRIPTOR.message_types_by_name['Model'] = _MODEL 683 | 684 | Pipeline = _reflection.GeneratedProtocolMessageType('Pipeline', (_message.Message,), dict( 685 | DESCRIPTOR = _PIPELINE, 686 | __module__ = 'Model_pb2' 687 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Pipeline) 688 | )) 689 | _sym_db.RegisterMessage(Pipeline) 690 | 691 | PipelineClassifier = _reflection.GeneratedProtocolMessageType('PipelineClassifier', (_message.Message,), dict( 692 | DESCRIPTOR = _PIPELINECLASSIFIER, 693 | __module__ = 'Model_pb2' 694 | # @@protoc_insertion_point(class_scope:CoreML.Specification.PipelineClassifier) 695 | )) 696 | _sym_db.RegisterMessage(PipelineClassifier) 697 | 698 | PipelineRegressor = _reflection.GeneratedProtocolMessageType('PipelineRegressor', (_message.Message,), dict( 699 | DESCRIPTOR = _PIPELINEREGRESSOR, 700 | __module__ = 'Model_pb2' 701 | # @@protoc_insertion_point(class_scope:CoreML.Specification.PipelineRegressor) 702 | )) 703 | _sym_db.RegisterMessage(PipelineRegressor) 704 | 705 | FeatureDescription = _reflection.GeneratedProtocolMessageType('FeatureDescription', (_message.Message,), dict( 706 | DESCRIPTOR = _FEATUREDESCRIPTION, 707 | __module__ = 'Model_pb2' 708 | # @@protoc_insertion_point(class_scope:CoreML.Specification.FeatureDescription) 709 | )) 710 | _sym_db.RegisterMessage(FeatureDescription) 711 | 712 | Metadata = _reflection.GeneratedProtocolMessageType('Metadata', (_message.Message,), dict( 713 | 714 | UserDefinedEntry = _reflection.GeneratedProtocolMessageType('UserDefinedEntry', (_message.Message,), dict( 715 | DESCRIPTOR = _METADATA_USERDEFINEDENTRY, 716 | __module__ = 'Model_pb2' 717 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Metadata.UserDefinedEntry) 718 | )) 719 | , 720 | DESCRIPTOR = _METADATA, 721 | __module__ = 'Model_pb2' 722 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Metadata) 723 | )) 724 | _sym_db.RegisterMessage(Metadata) 725 | _sym_db.RegisterMessage(Metadata.UserDefinedEntry) 726 | 727 | ModelDescription = _reflection.GeneratedProtocolMessageType('ModelDescription', (_message.Message,), dict( 728 | DESCRIPTOR = _MODELDESCRIPTION, 729 | __module__ = 'Model_pb2' 730 | # @@protoc_insertion_point(class_scope:CoreML.Specification.ModelDescription) 731 | )) 732 | _sym_db.RegisterMessage(ModelDescription) 733 | 734 | Model = _reflection.GeneratedProtocolMessageType('Model', (_message.Message,), dict( 735 | DESCRIPTOR = _MODEL, 736 | __module__ = 'Model_pb2' 737 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Model) 738 | )) 739 | _sym_db.RegisterMessage(Model) 740 | 741 | 742 | DESCRIPTOR.has_options = True 743 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 744 | _METADATA_USERDEFINEDENTRY.has_options = True 745 | _METADATA_USERDEFINEDENTRY._options = _descriptor._ParseOptions(descriptor_pb2.MessageOptions(), _b('8\001')) 746 | # @@protoc_insertion_point(module_scope) 747 | -------------------------------------------------------------------------------- /Normalizer_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: Normalizer.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='Normalizer.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x10Normalizer.proto\x12\x14\x43oreML.Specification\"o\n\nNormalizer\x12;\n\x08normType\x18\x01 \x01(\x0e\x32).CoreML.Specification.Normalizer.NormType\"$\n\x08NormType\x12\x08\n\x04LMax\x10\x00\x12\x06\n\x02L1\x10\x01\x12\x06\n\x02L2\x10\x02\x42\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | _NORMALIZER_NORMTYPE = _descriptor.EnumDescriptor( 29 | name='NormType', 30 | full_name='CoreML.Specification.Normalizer.NormType', 31 | filename=None, 32 | file=DESCRIPTOR, 33 | values=[ 34 | _descriptor.EnumValueDescriptor( 35 | name='LMax', index=0, number=0, 36 | options=None, 37 | type=None), 38 | _descriptor.EnumValueDescriptor( 39 | name='L1', index=1, number=1, 40 | options=None, 41 | type=None), 42 | _descriptor.EnumValueDescriptor( 43 | name='L2', index=2, number=2, 44 | options=None, 45 | type=None), 46 | ], 47 | containing_type=None, 48 | options=None, 49 | serialized_start=117, 50 | serialized_end=153, 51 | ) 52 | _sym_db.RegisterEnumDescriptor(_NORMALIZER_NORMTYPE) 53 | 54 | 55 | _NORMALIZER = _descriptor.Descriptor( 56 | name='Normalizer', 57 | full_name='CoreML.Specification.Normalizer', 58 | filename=None, 59 | file=DESCRIPTOR, 60 | containing_type=None, 61 | fields=[ 62 | _descriptor.FieldDescriptor( 63 | name='normType', full_name='CoreML.Specification.Normalizer.normType', index=0, 64 | number=1, type=14, cpp_type=8, label=1, 65 | has_default_value=False, default_value=0, 66 | message_type=None, enum_type=None, containing_type=None, 67 | is_extension=False, extension_scope=None, 68 | options=None), 69 | ], 70 | extensions=[ 71 | ], 72 | nested_types=[], 73 | enum_types=[ 74 | _NORMALIZER_NORMTYPE, 75 | ], 76 | options=None, 77 | is_extendable=False, 78 | syntax='proto3', 79 | extension_ranges=[], 80 | oneofs=[ 81 | ], 82 | serialized_start=42, 83 | serialized_end=153, 84 | ) 85 | 86 | _NORMALIZER.fields_by_name['normType'].enum_type = _NORMALIZER_NORMTYPE 87 | _NORMALIZER_NORMTYPE.containing_type = _NORMALIZER 88 | DESCRIPTOR.message_types_by_name['Normalizer'] = _NORMALIZER 89 | 90 | Normalizer = _reflection.GeneratedProtocolMessageType('Normalizer', (_message.Message,), dict( 91 | DESCRIPTOR = _NORMALIZER, 92 | __module__ = 'Normalizer_pb2' 93 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Normalizer) 94 | )) 95 | _sym_db.RegisterMessage(Normalizer) 96 | 97 | 98 | DESCRIPTOR.has_options = True 99 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 100 | # @@protoc_insertion_point(module_scope) 101 | -------------------------------------------------------------------------------- /OneHotEncoder_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: OneHotEncoder.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import DataStructures_pb2 as DataStructures__pb2 17 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 18 | 19 | from DataStructures_pb2 import * 20 | 21 | DESCRIPTOR = _descriptor.FileDescriptor( 22 | name='OneHotEncoder.proto', 23 | package='CoreML.Specification', 24 | syntax='proto3', 25 | serialized_pb=_b('\n\x13OneHotEncoder.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\xb5\x02\n\rOneHotEncoder\x12>\n\x10stringCategories\x18\x01 \x01(\x0b\x32\".CoreML.Specification.StringVectorH\x00\x12<\n\x0fint64Categories\x18\x02 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x00\x12\x14\n\x0coutputSparse\x18\n \x01(\x08\x12H\n\rhandleUnknown\x18\x0b \x01(\x0e\x32\x31.CoreML.Specification.OneHotEncoder.HandleUnknown\"6\n\rHandleUnknown\x12\x12\n\x0e\x45rrorOnUnknown\x10\x00\x12\x11\n\rIgnoreUnknown\x10\x01\x42\x0e\n\x0c\x43\x61tegoryTypeB\x02H\x03P\x00\x62\x06proto3') 26 | , 27 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 28 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 29 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 30 | 31 | 32 | 33 | _ONEHOTENCODER_HANDLEUNKNOWN = _descriptor.EnumDescriptor( 34 | name='HandleUnknown', 35 | full_name='CoreML.Specification.OneHotEncoder.HandleUnknown', 36 | filename=None, 37 | file=DESCRIPTOR, 38 | values=[ 39 | _descriptor.EnumValueDescriptor( 40 | name='ErrorOnUnknown', index=0, number=0, 41 | options=None, 42 | type=None), 43 | _descriptor.EnumValueDescriptor( 44 | name='IgnoreUnknown', index=1, number=1, 45 | options=None, 46 | type=None), 47 | ], 48 | containing_type=None, 49 | options=None, 50 | serialized_start=307, 51 | serialized_end=361, 52 | ) 53 | _sym_db.RegisterEnumDescriptor(_ONEHOTENCODER_HANDLEUNKNOWN) 54 | 55 | 56 | _ONEHOTENCODER = _descriptor.Descriptor( 57 | name='OneHotEncoder', 58 | full_name='CoreML.Specification.OneHotEncoder', 59 | filename=None, 60 | file=DESCRIPTOR, 61 | containing_type=None, 62 | fields=[ 63 | _descriptor.FieldDescriptor( 64 | name='stringCategories', full_name='CoreML.Specification.OneHotEncoder.stringCategories', index=0, 65 | number=1, type=11, cpp_type=10, label=1, 66 | has_default_value=False, default_value=None, 67 | message_type=None, enum_type=None, containing_type=None, 68 | is_extension=False, extension_scope=None, 69 | options=None), 70 | _descriptor.FieldDescriptor( 71 | name='int64Categories', full_name='CoreML.Specification.OneHotEncoder.int64Categories', index=1, 72 | number=2, type=11, cpp_type=10, label=1, 73 | has_default_value=False, default_value=None, 74 | message_type=None, enum_type=None, containing_type=None, 75 | is_extension=False, extension_scope=None, 76 | options=None), 77 | _descriptor.FieldDescriptor( 78 | name='outputSparse', full_name='CoreML.Specification.OneHotEncoder.outputSparse', index=2, 79 | number=10, type=8, cpp_type=7, label=1, 80 | has_default_value=False, default_value=False, 81 | message_type=None, enum_type=None, containing_type=None, 82 | is_extension=False, extension_scope=None, 83 | options=None), 84 | _descriptor.FieldDescriptor( 85 | name='handleUnknown', full_name='CoreML.Specification.OneHotEncoder.handleUnknown', index=3, 86 | number=11, type=14, cpp_type=8, label=1, 87 | has_default_value=False, default_value=0, 88 | message_type=None, enum_type=None, containing_type=None, 89 | is_extension=False, extension_scope=None, 90 | options=None), 91 | ], 92 | extensions=[ 93 | ], 94 | nested_types=[], 95 | enum_types=[ 96 | _ONEHOTENCODER_HANDLEUNKNOWN, 97 | ], 98 | options=None, 99 | is_extendable=False, 100 | syntax='proto3', 101 | extension_ranges=[], 102 | oneofs=[ 103 | _descriptor.OneofDescriptor( 104 | name='CategoryType', full_name='CoreML.Specification.OneHotEncoder.CategoryType', 105 | index=0, containing_type=None, fields=[]), 106 | ], 107 | serialized_start=68, 108 | serialized_end=377, 109 | ) 110 | 111 | _ONEHOTENCODER.fields_by_name['stringCategories'].message_type = DataStructures__pb2._STRINGVECTOR 112 | _ONEHOTENCODER.fields_by_name['int64Categories'].message_type = DataStructures__pb2._INT64VECTOR 113 | _ONEHOTENCODER.fields_by_name['handleUnknown'].enum_type = _ONEHOTENCODER_HANDLEUNKNOWN 114 | _ONEHOTENCODER_HANDLEUNKNOWN.containing_type = _ONEHOTENCODER 115 | _ONEHOTENCODER.oneofs_by_name['CategoryType'].fields.append( 116 | _ONEHOTENCODER.fields_by_name['stringCategories']) 117 | _ONEHOTENCODER.fields_by_name['stringCategories'].containing_oneof = _ONEHOTENCODER.oneofs_by_name['CategoryType'] 118 | _ONEHOTENCODER.oneofs_by_name['CategoryType'].fields.append( 119 | _ONEHOTENCODER.fields_by_name['int64Categories']) 120 | _ONEHOTENCODER.fields_by_name['int64Categories'].containing_oneof = _ONEHOTENCODER.oneofs_by_name['CategoryType'] 121 | DESCRIPTOR.message_types_by_name['OneHotEncoder'] = _ONEHOTENCODER 122 | 123 | OneHotEncoder = _reflection.GeneratedProtocolMessageType('OneHotEncoder', (_message.Message,), dict( 124 | DESCRIPTOR = _ONEHOTENCODER, 125 | __module__ = 'OneHotEncoder_pb2' 126 | # @@protoc_insertion_point(class_scope:CoreML.Specification.OneHotEncoder) 127 | )) 128 | _sym_db.RegisterMessage(OneHotEncoder) 129 | 130 | 131 | DESCRIPTOR.has_options = True 132 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 133 | # @@protoc_insertion_point(module_scope) 134 | -------------------------------------------------------------------------------- /README.markdown: -------------------------------------------------------------------------------- 1 | # Inside Core ML 2 | 3 | This is a Python script that lets you look inside your Core ML **.mlmodel** files. Useful for troubleshooting and to satisfy the curious mind. 4 | 5 | To use it: 6 | 7 | ``` 8 | python3 lookieloo.py name.mlmodel 9 | ``` 10 | 11 | This will print a bunch of information about your model to stdout. 12 | 13 | Requirements: 14 | 15 | - Python 3 16 | - protobuf 17 | 18 | **NOTE:** You will want to tweak the script to suit your own model. Right now it only prints information for models that are neural network classifiers (like Inception-v3). If you use a different type of model, you'll need to look at different fields. 19 | 20 | The mlmodel spec is written in protobuf, so to learn more about the possible fields in your model you should check out the [Core ML model specification](https://docs-assets.developer.apple.com/coreml/documentation/mlmodel_specification.zip). For example, the possible fields for a convolution layer are documented in NeuralNetwork.proto, under `ConvolutionLayerParams`. 21 | 22 | To see your entire model, enable the line: 23 | 24 | ```python 25 | print(model) 26 | ``` 27 | 28 | If you do this you should really pipe the output from the script to a file because it will include all the weights, so this is going to be a big data dump: 29 | 30 | ``` 31 | python3 lookieloo.py name.mlmodel > ~/Desktop/out.txt 32 | ``` 33 | 34 | The current version of this script is made for **coremltools 0.3.0**. If you have a later version of coremltools, you will need to regenerate the **_pb2.py** files. Here's how to do this: 35 | 36 | - [Download the latest Core ML model specification](https://docs-assets.developer.apple.com/coreml/documentation/mlmodel_specification.zip) 37 | - From the mlmodel_specification folder, run the following command: 38 | 39 | ``` 40 | protoc --python_out=/path/to/InsideCoreML *.proto 41 | ``` 42 | 43 | Links: [coremltools](https://pypi.python.org/pypi/coremltools) 44 | -------------------------------------------------------------------------------- /SVM_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: SVM.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | import DataStructures_pb2 as DataStructures__pb2 17 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 18 | 19 | from DataStructures_pb2 import * 20 | 21 | DESCRIPTOR = _descriptor.FileDescriptor( 22 | name='SVM.proto', 23 | package='CoreML.Specification', 24 | syntax='proto3', 25 | serialized_pb=_b('\n\tSVM.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\x0e\n\x0cLinearKernel\"\x1a\n\tRBFKernel\x12\r\n\x05gamma\x18\x01 \x01(\x01\"6\n\nPolyKernel\x12\x0e\n\x06\x64\x65gree\x18\x01 \x01(\x05\x12\t\n\x01\x63\x18\x02 \x01(\x01\x12\r\n\x05gamma\x18\x03 \x01(\x01\")\n\rSigmoidKernel\x12\r\n\x05gamma\x18\x01 \x01(\x01\x12\t\n\x01\x63\x18\x02 \x01(\x01\"\xfa\x01\n\x06Kernel\x12:\n\x0clinearKernel\x18\x01 \x01(\x0b\x32\".CoreML.Specification.LinearKernelH\x00\x12\x34\n\trbfKernel\x18\x02 \x01(\x0b\x32\x1f.CoreML.Specification.RBFKernelH\x00\x12\x36\n\npolyKernel\x18\x03 \x01(\x0b\x32 .CoreML.Specification.PolyKernelH\x00\x12<\n\rsigmoidKernel\x18\x04 \x01(\x0b\x32#.CoreML.Specification.SigmoidKernelH\x00\x42\x08\n\x06kernel\"*\n\nSparseNode\x12\r\n\x05index\x18\x01 \x01(\x05\x12\r\n\x05value\x18\x02 \x01(\x01\"?\n\x0cSparseVector\x12/\n\x05nodes\x18\x01 \x03(\x0b\x32 .CoreML.Specification.SparseNode\"K\n\x14SparseSupportVectors\x12\x33\n\x07vectors\x18\x01 \x03(\x0b\x32\".CoreML.Specification.SparseVector\"\x1d\n\x0b\x44\x65nseVector\x12\x0e\n\x06values\x18\x01 \x03(\x01\"I\n\x13\x44\x65nseSupportVectors\x12\x32\n\x07vectors\x18\x01 \x03(\x0b\x32!.CoreML.Specification.DenseVector\"\x1d\n\x0c\x43oefficients\x12\r\n\x05\x61lpha\x18\x01 \x03(\x01\"\xb5\x02\n\x16SupportVectorRegressor\x12,\n\x06kernel\x18\x01 \x01(\x0b\x32\x1c.CoreML.Specification.Kernel\x12J\n\x14sparseSupportVectors\x18\x02 \x01(\x0b\x32*.CoreML.Specification.SparseSupportVectorsH\x00\x12H\n\x13\x64\x65nseSupportVectors\x18\x03 \x01(\x0b\x32).CoreML.Specification.DenseSupportVectorsH\x00\x12\x38\n\x0c\x63oefficients\x18\x04 \x01(\x0b\x32\".CoreML.Specification.Coefficients\x12\x0b\n\x03rho\x18\x05 \x01(\x01\x42\x10\n\x0esupportVectors\"\x8b\x04\n\x17SupportVectorClassifier\x12,\n\x06kernel\x18\x01 \x01(\x0b\x32\x1c.CoreML.Specification.Kernel\x12&\n\x1enumberOfSupportVectorsPerClass\x18\x02 \x03(\x05\x12J\n\x14sparseSupportVectors\x18\x03 \x01(\x0b\x32*.CoreML.Specification.SparseSupportVectorsH\x00\x12H\n\x13\x64\x65nseSupportVectors\x18\x04 \x01(\x0b\x32).CoreML.Specification.DenseSupportVectorsH\x00\x12\x38\n\x0c\x63oefficients\x18\x05 \x03(\x0b\x32\".CoreML.Specification.Coefficients\x12\x0b\n\x03rho\x18\x06 \x03(\x01\x12\r\n\x05probA\x18\x07 \x03(\x01\x12\r\n\x05probB\x18\x08 \x03(\x01\x12?\n\x11stringClassLabels\x18\x64 \x01(\x0b\x32\".CoreML.Specification.StringVectorH\x01\x12=\n\x10int64ClassLabels\x18\x65 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x01\x42\x10\n\x0esupportVectorsB\r\n\x0b\x43lassLabelsB\x02H\x03P\x00\x62\x06proto3') 26 | , 27 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 28 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 29 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 30 | 31 | 32 | 33 | 34 | _LINEARKERNEL = _descriptor.Descriptor( 35 | name='LinearKernel', 36 | full_name='CoreML.Specification.LinearKernel', 37 | filename=None, 38 | file=DESCRIPTOR, 39 | containing_type=None, 40 | fields=[ 41 | ], 42 | extensions=[ 43 | ], 44 | nested_types=[], 45 | enum_types=[ 46 | ], 47 | options=None, 48 | is_extendable=False, 49 | syntax='proto3', 50 | extension_ranges=[], 51 | oneofs=[ 52 | ], 53 | serialized_start=57, 54 | serialized_end=71, 55 | ) 56 | 57 | 58 | _RBFKERNEL = _descriptor.Descriptor( 59 | name='RBFKernel', 60 | full_name='CoreML.Specification.RBFKernel', 61 | filename=None, 62 | file=DESCRIPTOR, 63 | containing_type=None, 64 | fields=[ 65 | _descriptor.FieldDescriptor( 66 | name='gamma', full_name='CoreML.Specification.RBFKernel.gamma', index=0, 67 | number=1, type=1, cpp_type=5, label=1, 68 | has_default_value=False, default_value=float(0), 69 | message_type=None, enum_type=None, containing_type=None, 70 | is_extension=False, extension_scope=None, 71 | options=None), 72 | ], 73 | extensions=[ 74 | ], 75 | nested_types=[], 76 | enum_types=[ 77 | ], 78 | options=None, 79 | is_extendable=False, 80 | syntax='proto3', 81 | extension_ranges=[], 82 | oneofs=[ 83 | ], 84 | serialized_start=73, 85 | serialized_end=99, 86 | ) 87 | 88 | 89 | _POLYKERNEL = _descriptor.Descriptor( 90 | name='PolyKernel', 91 | full_name='CoreML.Specification.PolyKernel', 92 | filename=None, 93 | file=DESCRIPTOR, 94 | containing_type=None, 95 | fields=[ 96 | _descriptor.FieldDescriptor( 97 | name='degree', full_name='CoreML.Specification.PolyKernel.degree', index=0, 98 | number=1, type=5, cpp_type=1, label=1, 99 | has_default_value=False, default_value=0, 100 | message_type=None, enum_type=None, containing_type=None, 101 | is_extension=False, extension_scope=None, 102 | options=None), 103 | _descriptor.FieldDescriptor( 104 | name='c', full_name='CoreML.Specification.PolyKernel.c', index=1, 105 | number=2, type=1, cpp_type=5, label=1, 106 | has_default_value=False, default_value=float(0), 107 | message_type=None, enum_type=None, containing_type=None, 108 | is_extension=False, extension_scope=None, 109 | options=None), 110 | _descriptor.FieldDescriptor( 111 | name='gamma', full_name='CoreML.Specification.PolyKernel.gamma', index=2, 112 | number=3, type=1, cpp_type=5, label=1, 113 | has_default_value=False, default_value=float(0), 114 | message_type=None, enum_type=None, containing_type=None, 115 | is_extension=False, extension_scope=None, 116 | options=None), 117 | ], 118 | extensions=[ 119 | ], 120 | nested_types=[], 121 | enum_types=[ 122 | ], 123 | options=None, 124 | is_extendable=False, 125 | syntax='proto3', 126 | extension_ranges=[], 127 | oneofs=[ 128 | ], 129 | serialized_start=101, 130 | serialized_end=155, 131 | ) 132 | 133 | 134 | _SIGMOIDKERNEL = _descriptor.Descriptor( 135 | name='SigmoidKernel', 136 | full_name='CoreML.Specification.SigmoidKernel', 137 | filename=None, 138 | file=DESCRIPTOR, 139 | containing_type=None, 140 | fields=[ 141 | _descriptor.FieldDescriptor( 142 | name='gamma', full_name='CoreML.Specification.SigmoidKernel.gamma', index=0, 143 | number=1, type=1, cpp_type=5, label=1, 144 | has_default_value=False, default_value=float(0), 145 | message_type=None, enum_type=None, containing_type=None, 146 | is_extension=False, extension_scope=None, 147 | options=None), 148 | _descriptor.FieldDescriptor( 149 | name='c', full_name='CoreML.Specification.SigmoidKernel.c', index=1, 150 | number=2, type=1, cpp_type=5, label=1, 151 | has_default_value=False, default_value=float(0), 152 | message_type=None, enum_type=None, containing_type=None, 153 | is_extension=False, extension_scope=None, 154 | options=None), 155 | ], 156 | extensions=[ 157 | ], 158 | nested_types=[], 159 | enum_types=[ 160 | ], 161 | options=None, 162 | is_extendable=False, 163 | syntax='proto3', 164 | extension_ranges=[], 165 | oneofs=[ 166 | ], 167 | serialized_start=157, 168 | serialized_end=198, 169 | ) 170 | 171 | 172 | _KERNEL = _descriptor.Descriptor( 173 | name='Kernel', 174 | full_name='CoreML.Specification.Kernel', 175 | filename=None, 176 | file=DESCRIPTOR, 177 | containing_type=None, 178 | fields=[ 179 | _descriptor.FieldDescriptor( 180 | name='linearKernel', full_name='CoreML.Specification.Kernel.linearKernel', index=0, 181 | number=1, type=11, cpp_type=10, label=1, 182 | has_default_value=False, default_value=None, 183 | message_type=None, enum_type=None, containing_type=None, 184 | is_extension=False, extension_scope=None, 185 | options=None), 186 | _descriptor.FieldDescriptor( 187 | name='rbfKernel', full_name='CoreML.Specification.Kernel.rbfKernel', index=1, 188 | number=2, type=11, cpp_type=10, label=1, 189 | has_default_value=False, default_value=None, 190 | message_type=None, enum_type=None, containing_type=None, 191 | is_extension=False, extension_scope=None, 192 | options=None), 193 | _descriptor.FieldDescriptor( 194 | name='polyKernel', full_name='CoreML.Specification.Kernel.polyKernel', index=2, 195 | number=3, type=11, cpp_type=10, label=1, 196 | has_default_value=False, default_value=None, 197 | message_type=None, enum_type=None, containing_type=None, 198 | is_extension=False, extension_scope=None, 199 | options=None), 200 | _descriptor.FieldDescriptor( 201 | name='sigmoidKernel', full_name='CoreML.Specification.Kernel.sigmoidKernel', index=3, 202 | number=4, type=11, cpp_type=10, label=1, 203 | has_default_value=False, default_value=None, 204 | message_type=None, enum_type=None, containing_type=None, 205 | is_extension=False, extension_scope=None, 206 | options=None), 207 | ], 208 | extensions=[ 209 | ], 210 | nested_types=[], 211 | enum_types=[ 212 | ], 213 | options=None, 214 | is_extendable=False, 215 | syntax='proto3', 216 | extension_ranges=[], 217 | oneofs=[ 218 | _descriptor.OneofDescriptor( 219 | name='kernel', full_name='CoreML.Specification.Kernel.kernel', 220 | index=0, containing_type=None, fields=[]), 221 | ], 222 | serialized_start=201, 223 | serialized_end=451, 224 | ) 225 | 226 | 227 | _SPARSENODE = _descriptor.Descriptor( 228 | name='SparseNode', 229 | full_name='CoreML.Specification.SparseNode', 230 | filename=None, 231 | file=DESCRIPTOR, 232 | containing_type=None, 233 | fields=[ 234 | _descriptor.FieldDescriptor( 235 | name='index', full_name='CoreML.Specification.SparseNode.index', index=0, 236 | number=1, type=5, cpp_type=1, label=1, 237 | has_default_value=False, default_value=0, 238 | message_type=None, enum_type=None, containing_type=None, 239 | is_extension=False, extension_scope=None, 240 | options=None), 241 | _descriptor.FieldDescriptor( 242 | name='value', full_name='CoreML.Specification.SparseNode.value', index=1, 243 | number=2, type=1, cpp_type=5, label=1, 244 | has_default_value=False, default_value=float(0), 245 | message_type=None, enum_type=None, containing_type=None, 246 | is_extension=False, extension_scope=None, 247 | options=None), 248 | ], 249 | extensions=[ 250 | ], 251 | nested_types=[], 252 | enum_types=[ 253 | ], 254 | options=None, 255 | is_extendable=False, 256 | syntax='proto3', 257 | extension_ranges=[], 258 | oneofs=[ 259 | ], 260 | serialized_start=453, 261 | serialized_end=495, 262 | ) 263 | 264 | 265 | _SPARSEVECTOR = _descriptor.Descriptor( 266 | name='SparseVector', 267 | full_name='CoreML.Specification.SparseVector', 268 | filename=None, 269 | file=DESCRIPTOR, 270 | containing_type=None, 271 | fields=[ 272 | _descriptor.FieldDescriptor( 273 | name='nodes', full_name='CoreML.Specification.SparseVector.nodes', index=0, 274 | number=1, type=11, cpp_type=10, label=3, 275 | has_default_value=False, default_value=[], 276 | message_type=None, enum_type=None, containing_type=None, 277 | is_extension=False, extension_scope=None, 278 | options=None), 279 | ], 280 | extensions=[ 281 | ], 282 | nested_types=[], 283 | enum_types=[ 284 | ], 285 | options=None, 286 | is_extendable=False, 287 | syntax='proto3', 288 | extension_ranges=[], 289 | oneofs=[ 290 | ], 291 | serialized_start=497, 292 | serialized_end=560, 293 | ) 294 | 295 | 296 | _SPARSESUPPORTVECTORS = _descriptor.Descriptor( 297 | name='SparseSupportVectors', 298 | full_name='CoreML.Specification.SparseSupportVectors', 299 | filename=None, 300 | file=DESCRIPTOR, 301 | containing_type=None, 302 | fields=[ 303 | _descriptor.FieldDescriptor( 304 | name='vectors', full_name='CoreML.Specification.SparseSupportVectors.vectors', index=0, 305 | number=1, type=11, cpp_type=10, label=3, 306 | has_default_value=False, default_value=[], 307 | message_type=None, enum_type=None, containing_type=None, 308 | is_extension=False, extension_scope=None, 309 | options=None), 310 | ], 311 | extensions=[ 312 | ], 313 | nested_types=[], 314 | enum_types=[ 315 | ], 316 | options=None, 317 | is_extendable=False, 318 | syntax='proto3', 319 | extension_ranges=[], 320 | oneofs=[ 321 | ], 322 | serialized_start=562, 323 | serialized_end=637, 324 | ) 325 | 326 | 327 | _DENSEVECTOR = _descriptor.Descriptor( 328 | name='DenseVector', 329 | full_name='CoreML.Specification.DenseVector', 330 | filename=None, 331 | file=DESCRIPTOR, 332 | containing_type=None, 333 | fields=[ 334 | _descriptor.FieldDescriptor( 335 | name='values', full_name='CoreML.Specification.DenseVector.values', index=0, 336 | number=1, type=1, cpp_type=5, label=3, 337 | has_default_value=False, default_value=[], 338 | message_type=None, enum_type=None, containing_type=None, 339 | is_extension=False, extension_scope=None, 340 | options=None), 341 | ], 342 | extensions=[ 343 | ], 344 | nested_types=[], 345 | enum_types=[ 346 | ], 347 | options=None, 348 | is_extendable=False, 349 | syntax='proto3', 350 | extension_ranges=[], 351 | oneofs=[ 352 | ], 353 | serialized_start=639, 354 | serialized_end=668, 355 | ) 356 | 357 | 358 | _DENSESUPPORTVECTORS = _descriptor.Descriptor( 359 | name='DenseSupportVectors', 360 | full_name='CoreML.Specification.DenseSupportVectors', 361 | filename=None, 362 | file=DESCRIPTOR, 363 | containing_type=None, 364 | fields=[ 365 | _descriptor.FieldDescriptor( 366 | name='vectors', full_name='CoreML.Specification.DenseSupportVectors.vectors', index=0, 367 | number=1, type=11, cpp_type=10, label=3, 368 | has_default_value=False, default_value=[], 369 | message_type=None, enum_type=None, containing_type=None, 370 | is_extension=False, extension_scope=None, 371 | options=None), 372 | ], 373 | extensions=[ 374 | ], 375 | nested_types=[], 376 | enum_types=[ 377 | ], 378 | options=None, 379 | is_extendable=False, 380 | syntax='proto3', 381 | extension_ranges=[], 382 | oneofs=[ 383 | ], 384 | serialized_start=670, 385 | serialized_end=743, 386 | ) 387 | 388 | 389 | _COEFFICIENTS = _descriptor.Descriptor( 390 | name='Coefficients', 391 | full_name='CoreML.Specification.Coefficients', 392 | filename=None, 393 | file=DESCRIPTOR, 394 | containing_type=None, 395 | fields=[ 396 | _descriptor.FieldDescriptor( 397 | name='alpha', full_name='CoreML.Specification.Coefficients.alpha', index=0, 398 | number=1, type=1, cpp_type=5, label=3, 399 | has_default_value=False, default_value=[], 400 | message_type=None, enum_type=None, containing_type=None, 401 | is_extension=False, extension_scope=None, 402 | options=None), 403 | ], 404 | extensions=[ 405 | ], 406 | nested_types=[], 407 | enum_types=[ 408 | ], 409 | options=None, 410 | is_extendable=False, 411 | syntax='proto3', 412 | extension_ranges=[], 413 | oneofs=[ 414 | ], 415 | serialized_start=745, 416 | serialized_end=774, 417 | ) 418 | 419 | 420 | _SUPPORTVECTORREGRESSOR = _descriptor.Descriptor( 421 | name='SupportVectorRegressor', 422 | full_name='CoreML.Specification.SupportVectorRegressor', 423 | filename=None, 424 | file=DESCRIPTOR, 425 | containing_type=None, 426 | fields=[ 427 | _descriptor.FieldDescriptor( 428 | name='kernel', full_name='CoreML.Specification.SupportVectorRegressor.kernel', index=0, 429 | number=1, type=11, cpp_type=10, label=1, 430 | has_default_value=False, default_value=None, 431 | message_type=None, enum_type=None, containing_type=None, 432 | is_extension=False, extension_scope=None, 433 | options=None), 434 | _descriptor.FieldDescriptor( 435 | name='sparseSupportVectors', full_name='CoreML.Specification.SupportVectorRegressor.sparseSupportVectors', index=1, 436 | number=2, type=11, cpp_type=10, label=1, 437 | has_default_value=False, default_value=None, 438 | message_type=None, enum_type=None, containing_type=None, 439 | is_extension=False, extension_scope=None, 440 | options=None), 441 | _descriptor.FieldDescriptor( 442 | name='denseSupportVectors', full_name='CoreML.Specification.SupportVectorRegressor.denseSupportVectors', index=2, 443 | number=3, type=11, cpp_type=10, label=1, 444 | has_default_value=False, default_value=None, 445 | message_type=None, enum_type=None, containing_type=None, 446 | is_extension=False, extension_scope=None, 447 | options=None), 448 | _descriptor.FieldDescriptor( 449 | name='coefficients', full_name='CoreML.Specification.SupportVectorRegressor.coefficients', index=3, 450 | number=4, type=11, cpp_type=10, label=1, 451 | has_default_value=False, default_value=None, 452 | message_type=None, enum_type=None, containing_type=None, 453 | is_extension=False, extension_scope=None, 454 | options=None), 455 | _descriptor.FieldDescriptor( 456 | name='rho', full_name='CoreML.Specification.SupportVectorRegressor.rho', index=4, 457 | number=5, type=1, cpp_type=5, label=1, 458 | has_default_value=False, default_value=float(0), 459 | message_type=None, enum_type=None, containing_type=None, 460 | is_extension=False, extension_scope=None, 461 | options=None), 462 | ], 463 | extensions=[ 464 | ], 465 | nested_types=[], 466 | enum_types=[ 467 | ], 468 | options=None, 469 | is_extendable=False, 470 | syntax='proto3', 471 | extension_ranges=[], 472 | oneofs=[ 473 | _descriptor.OneofDescriptor( 474 | name='supportVectors', full_name='CoreML.Specification.SupportVectorRegressor.supportVectors', 475 | index=0, containing_type=None, fields=[]), 476 | ], 477 | serialized_start=777, 478 | serialized_end=1086, 479 | ) 480 | 481 | 482 | _SUPPORTVECTORCLASSIFIER = _descriptor.Descriptor( 483 | name='SupportVectorClassifier', 484 | full_name='CoreML.Specification.SupportVectorClassifier', 485 | filename=None, 486 | file=DESCRIPTOR, 487 | containing_type=None, 488 | fields=[ 489 | _descriptor.FieldDescriptor( 490 | name='kernel', full_name='CoreML.Specification.SupportVectorClassifier.kernel', index=0, 491 | number=1, type=11, cpp_type=10, label=1, 492 | has_default_value=False, default_value=None, 493 | message_type=None, enum_type=None, containing_type=None, 494 | is_extension=False, extension_scope=None, 495 | options=None), 496 | _descriptor.FieldDescriptor( 497 | name='numberOfSupportVectorsPerClass', full_name='CoreML.Specification.SupportVectorClassifier.numberOfSupportVectorsPerClass', index=1, 498 | number=2, type=5, cpp_type=1, label=3, 499 | has_default_value=False, default_value=[], 500 | message_type=None, enum_type=None, containing_type=None, 501 | is_extension=False, extension_scope=None, 502 | options=None), 503 | _descriptor.FieldDescriptor( 504 | name='sparseSupportVectors', full_name='CoreML.Specification.SupportVectorClassifier.sparseSupportVectors', index=2, 505 | number=3, type=11, cpp_type=10, label=1, 506 | has_default_value=False, default_value=None, 507 | message_type=None, enum_type=None, containing_type=None, 508 | is_extension=False, extension_scope=None, 509 | options=None), 510 | _descriptor.FieldDescriptor( 511 | name='denseSupportVectors', full_name='CoreML.Specification.SupportVectorClassifier.denseSupportVectors', index=3, 512 | number=4, type=11, cpp_type=10, label=1, 513 | has_default_value=False, default_value=None, 514 | message_type=None, enum_type=None, containing_type=None, 515 | is_extension=False, extension_scope=None, 516 | options=None), 517 | _descriptor.FieldDescriptor( 518 | name='coefficients', full_name='CoreML.Specification.SupportVectorClassifier.coefficients', index=4, 519 | number=5, type=11, cpp_type=10, label=3, 520 | has_default_value=False, default_value=[], 521 | message_type=None, enum_type=None, containing_type=None, 522 | is_extension=False, extension_scope=None, 523 | options=None), 524 | _descriptor.FieldDescriptor( 525 | name='rho', full_name='CoreML.Specification.SupportVectorClassifier.rho', index=5, 526 | number=6, type=1, cpp_type=5, label=3, 527 | has_default_value=False, default_value=[], 528 | message_type=None, enum_type=None, containing_type=None, 529 | is_extension=False, extension_scope=None, 530 | options=None), 531 | _descriptor.FieldDescriptor( 532 | name='probA', full_name='CoreML.Specification.SupportVectorClassifier.probA', index=6, 533 | number=7, type=1, cpp_type=5, label=3, 534 | has_default_value=False, default_value=[], 535 | message_type=None, enum_type=None, containing_type=None, 536 | is_extension=False, extension_scope=None, 537 | options=None), 538 | _descriptor.FieldDescriptor( 539 | name='probB', full_name='CoreML.Specification.SupportVectorClassifier.probB', index=7, 540 | number=8, type=1, cpp_type=5, label=3, 541 | has_default_value=False, default_value=[], 542 | message_type=None, enum_type=None, containing_type=None, 543 | is_extension=False, extension_scope=None, 544 | options=None), 545 | _descriptor.FieldDescriptor( 546 | name='stringClassLabels', full_name='CoreML.Specification.SupportVectorClassifier.stringClassLabels', index=8, 547 | number=100, type=11, cpp_type=10, label=1, 548 | has_default_value=False, default_value=None, 549 | message_type=None, enum_type=None, containing_type=None, 550 | is_extension=False, extension_scope=None, 551 | options=None), 552 | _descriptor.FieldDescriptor( 553 | name='int64ClassLabels', full_name='CoreML.Specification.SupportVectorClassifier.int64ClassLabels', index=9, 554 | number=101, type=11, cpp_type=10, label=1, 555 | has_default_value=False, default_value=None, 556 | message_type=None, enum_type=None, containing_type=None, 557 | is_extension=False, extension_scope=None, 558 | options=None), 559 | ], 560 | extensions=[ 561 | ], 562 | nested_types=[], 563 | enum_types=[ 564 | ], 565 | options=None, 566 | is_extendable=False, 567 | syntax='proto3', 568 | extension_ranges=[], 569 | oneofs=[ 570 | _descriptor.OneofDescriptor( 571 | name='supportVectors', full_name='CoreML.Specification.SupportVectorClassifier.supportVectors', 572 | index=0, containing_type=None, fields=[]), 573 | _descriptor.OneofDescriptor( 574 | name='ClassLabels', full_name='CoreML.Specification.SupportVectorClassifier.ClassLabels', 575 | index=1, containing_type=None, fields=[]), 576 | ], 577 | serialized_start=1089, 578 | serialized_end=1612, 579 | ) 580 | 581 | _KERNEL.fields_by_name['linearKernel'].message_type = _LINEARKERNEL 582 | _KERNEL.fields_by_name['rbfKernel'].message_type = _RBFKERNEL 583 | _KERNEL.fields_by_name['polyKernel'].message_type = _POLYKERNEL 584 | _KERNEL.fields_by_name['sigmoidKernel'].message_type = _SIGMOIDKERNEL 585 | _KERNEL.oneofs_by_name['kernel'].fields.append( 586 | _KERNEL.fields_by_name['linearKernel']) 587 | _KERNEL.fields_by_name['linearKernel'].containing_oneof = _KERNEL.oneofs_by_name['kernel'] 588 | _KERNEL.oneofs_by_name['kernel'].fields.append( 589 | _KERNEL.fields_by_name['rbfKernel']) 590 | _KERNEL.fields_by_name['rbfKernel'].containing_oneof = _KERNEL.oneofs_by_name['kernel'] 591 | _KERNEL.oneofs_by_name['kernel'].fields.append( 592 | _KERNEL.fields_by_name['polyKernel']) 593 | _KERNEL.fields_by_name['polyKernel'].containing_oneof = _KERNEL.oneofs_by_name['kernel'] 594 | _KERNEL.oneofs_by_name['kernel'].fields.append( 595 | _KERNEL.fields_by_name['sigmoidKernel']) 596 | _KERNEL.fields_by_name['sigmoidKernel'].containing_oneof = _KERNEL.oneofs_by_name['kernel'] 597 | _SPARSEVECTOR.fields_by_name['nodes'].message_type = _SPARSENODE 598 | _SPARSESUPPORTVECTORS.fields_by_name['vectors'].message_type = _SPARSEVECTOR 599 | _DENSESUPPORTVECTORS.fields_by_name['vectors'].message_type = _DENSEVECTOR 600 | _SUPPORTVECTORREGRESSOR.fields_by_name['kernel'].message_type = _KERNEL 601 | _SUPPORTVECTORREGRESSOR.fields_by_name['sparseSupportVectors'].message_type = _SPARSESUPPORTVECTORS 602 | _SUPPORTVECTORREGRESSOR.fields_by_name['denseSupportVectors'].message_type = _DENSESUPPORTVECTORS 603 | _SUPPORTVECTORREGRESSOR.fields_by_name['coefficients'].message_type = _COEFFICIENTS 604 | _SUPPORTVECTORREGRESSOR.oneofs_by_name['supportVectors'].fields.append( 605 | _SUPPORTVECTORREGRESSOR.fields_by_name['sparseSupportVectors']) 606 | _SUPPORTVECTORREGRESSOR.fields_by_name['sparseSupportVectors'].containing_oneof = _SUPPORTVECTORREGRESSOR.oneofs_by_name['supportVectors'] 607 | _SUPPORTVECTORREGRESSOR.oneofs_by_name['supportVectors'].fields.append( 608 | _SUPPORTVECTORREGRESSOR.fields_by_name['denseSupportVectors']) 609 | _SUPPORTVECTORREGRESSOR.fields_by_name['denseSupportVectors'].containing_oneof = _SUPPORTVECTORREGRESSOR.oneofs_by_name['supportVectors'] 610 | _SUPPORTVECTORCLASSIFIER.fields_by_name['kernel'].message_type = _KERNEL 611 | _SUPPORTVECTORCLASSIFIER.fields_by_name['sparseSupportVectors'].message_type = _SPARSESUPPORTVECTORS 612 | _SUPPORTVECTORCLASSIFIER.fields_by_name['denseSupportVectors'].message_type = _DENSESUPPORTVECTORS 613 | _SUPPORTVECTORCLASSIFIER.fields_by_name['coefficients'].message_type = _COEFFICIENTS 614 | _SUPPORTVECTORCLASSIFIER.fields_by_name['stringClassLabels'].message_type = DataStructures__pb2._STRINGVECTOR 615 | _SUPPORTVECTORCLASSIFIER.fields_by_name['int64ClassLabels'].message_type = DataStructures__pb2._INT64VECTOR 616 | _SUPPORTVECTORCLASSIFIER.oneofs_by_name['supportVectors'].fields.append( 617 | _SUPPORTVECTORCLASSIFIER.fields_by_name['sparseSupportVectors']) 618 | _SUPPORTVECTORCLASSIFIER.fields_by_name['sparseSupportVectors'].containing_oneof = _SUPPORTVECTORCLASSIFIER.oneofs_by_name['supportVectors'] 619 | _SUPPORTVECTORCLASSIFIER.oneofs_by_name['supportVectors'].fields.append( 620 | _SUPPORTVECTORCLASSIFIER.fields_by_name['denseSupportVectors']) 621 | _SUPPORTVECTORCLASSIFIER.fields_by_name['denseSupportVectors'].containing_oneof = _SUPPORTVECTORCLASSIFIER.oneofs_by_name['supportVectors'] 622 | _SUPPORTVECTORCLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 623 | _SUPPORTVECTORCLASSIFIER.fields_by_name['stringClassLabels']) 624 | _SUPPORTVECTORCLASSIFIER.fields_by_name['stringClassLabels'].containing_oneof = _SUPPORTVECTORCLASSIFIER.oneofs_by_name['ClassLabels'] 625 | _SUPPORTVECTORCLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 626 | _SUPPORTVECTORCLASSIFIER.fields_by_name['int64ClassLabels']) 627 | _SUPPORTVECTORCLASSIFIER.fields_by_name['int64ClassLabels'].containing_oneof = _SUPPORTVECTORCLASSIFIER.oneofs_by_name['ClassLabels'] 628 | DESCRIPTOR.message_types_by_name['LinearKernel'] = _LINEARKERNEL 629 | DESCRIPTOR.message_types_by_name['RBFKernel'] = _RBFKERNEL 630 | DESCRIPTOR.message_types_by_name['PolyKernel'] = _POLYKERNEL 631 | DESCRIPTOR.message_types_by_name['SigmoidKernel'] = _SIGMOIDKERNEL 632 | DESCRIPTOR.message_types_by_name['Kernel'] = _KERNEL 633 | DESCRIPTOR.message_types_by_name['SparseNode'] = _SPARSENODE 634 | DESCRIPTOR.message_types_by_name['SparseVector'] = _SPARSEVECTOR 635 | DESCRIPTOR.message_types_by_name['SparseSupportVectors'] = _SPARSESUPPORTVECTORS 636 | DESCRIPTOR.message_types_by_name['DenseVector'] = _DENSEVECTOR 637 | DESCRIPTOR.message_types_by_name['DenseSupportVectors'] = _DENSESUPPORTVECTORS 638 | DESCRIPTOR.message_types_by_name['Coefficients'] = _COEFFICIENTS 639 | DESCRIPTOR.message_types_by_name['SupportVectorRegressor'] = _SUPPORTVECTORREGRESSOR 640 | DESCRIPTOR.message_types_by_name['SupportVectorClassifier'] = _SUPPORTVECTORCLASSIFIER 641 | 642 | LinearKernel = _reflection.GeneratedProtocolMessageType('LinearKernel', (_message.Message,), dict( 643 | DESCRIPTOR = _LINEARKERNEL, 644 | __module__ = 'SVM_pb2' 645 | # @@protoc_insertion_point(class_scope:CoreML.Specification.LinearKernel) 646 | )) 647 | _sym_db.RegisterMessage(LinearKernel) 648 | 649 | RBFKernel = _reflection.GeneratedProtocolMessageType('RBFKernel', (_message.Message,), dict( 650 | DESCRIPTOR = _RBFKERNEL, 651 | __module__ = 'SVM_pb2' 652 | # @@protoc_insertion_point(class_scope:CoreML.Specification.RBFKernel) 653 | )) 654 | _sym_db.RegisterMessage(RBFKernel) 655 | 656 | PolyKernel = _reflection.GeneratedProtocolMessageType('PolyKernel', (_message.Message,), dict( 657 | DESCRIPTOR = _POLYKERNEL, 658 | __module__ = 'SVM_pb2' 659 | # @@protoc_insertion_point(class_scope:CoreML.Specification.PolyKernel) 660 | )) 661 | _sym_db.RegisterMessage(PolyKernel) 662 | 663 | SigmoidKernel = _reflection.GeneratedProtocolMessageType('SigmoidKernel', (_message.Message,), dict( 664 | DESCRIPTOR = _SIGMOIDKERNEL, 665 | __module__ = 'SVM_pb2' 666 | # @@protoc_insertion_point(class_scope:CoreML.Specification.SigmoidKernel) 667 | )) 668 | _sym_db.RegisterMessage(SigmoidKernel) 669 | 670 | Kernel = _reflection.GeneratedProtocolMessageType('Kernel', (_message.Message,), dict( 671 | DESCRIPTOR = _KERNEL, 672 | __module__ = 'SVM_pb2' 673 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Kernel) 674 | )) 675 | _sym_db.RegisterMessage(Kernel) 676 | 677 | SparseNode = _reflection.GeneratedProtocolMessageType('SparseNode', (_message.Message,), dict( 678 | DESCRIPTOR = _SPARSENODE, 679 | __module__ = 'SVM_pb2' 680 | # @@protoc_insertion_point(class_scope:CoreML.Specification.SparseNode) 681 | )) 682 | _sym_db.RegisterMessage(SparseNode) 683 | 684 | SparseVector = _reflection.GeneratedProtocolMessageType('SparseVector', (_message.Message,), dict( 685 | DESCRIPTOR = _SPARSEVECTOR, 686 | __module__ = 'SVM_pb2' 687 | # @@protoc_insertion_point(class_scope:CoreML.Specification.SparseVector) 688 | )) 689 | _sym_db.RegisterMessage(SparseVector) 690 | 691 | SparseSupportVectors = _reflection.GeneratedProtocolMessageType('SparseSupportVectors', (_message.Message,), dict( 692 | DESCRIPTOR = _SPARSESUPPORTVECTORS, 693 | __module__ = 'SVM_pb2' 694 | # @@protoc_insertion_point(class_scope:CoreML.Specification.SparseSupportVectors) 695 | )) 696 | _sym_db.RegisterMessage(SparseSupportVectors) 697 | 698 | DenseVector = _reflection.GeneratedProtocolMessageType('DenseVector', (_message.Message,), dict( 699 | DESCRIPTOR = _DENSEVECTOR, 700 | __module__ = 'SVM_pb2' 701 | # @@protoc_insertion_point(class_scope:CoreML.Specification.DenseVector) 702 | )) 703 | _sym_db.RegisterMessage(DenseVector) 704 | 705 | DenseSupportVectors = _reflection.GeneratedProtocolMessageType('DenseSupportVectors', (_message.Message,), dict( 706 | DESCRIPTOR = _DENSESUPPORTVECTORS, 707 | __module__ = 'SVM_pb2' 708 | # @@protoc_insertion_point(class_scope:CoreML.Specification.DenseSupportVectors) 709 | )) 710 | _sym_db.RegisterMessage(DenseSupportVectors) 711 | 712 | Coefficients = _reflection.GeneratedProtocolMessageType('Coefficients', (_message.Message,), dict( 713 | DESCRIPTOR = _COEFFICIENTS, 714 | __module__ = 'SVM_pb2' 715 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Coefficients) 716 | )) 717 | _sym_db.RegisterMessage(Coefficients) 718 | 719 | SupportVectorRegressor = _reflection.GeneratedProtocolMessageType('SupportVectorRegressor', (_message.Message,), dict( 720 | DESCRIPTOR = _SUPPORTVECTORREGRESSOR, 721 | __module__ = 'SVM_pb2' 722 | # @@protoc_insertion_point(class_scope:CoreML.Specification.SupportVectorRegressor) 723 | )) 724 | _sym_db.RegisterMessage(SupportVectorRegressor) 725 | 726 | SupportVectorClassifier = _reflection.GeneratedProtocolMessageType('SupportVectorClassifier', (_message.Message,), dict( 727 | DESCRIPTOR = _SUPPORTVECTORCLASSIFIER, 728 | __module__ = 'SVM_pb2' 729 | # @@protoc_insertion_point(class_scope:CoreML.Specification.SupportVectorClassifier) 730 | )) 731 | _sym_db.RegisterMessage(SupportVectorClassifier) 732 | 733 | 734 | DESCRIPTOR.has_options = True 735 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 736 | # @@protoc_insertion_point(module_scope) 737 | -------------------------------------------------------------------------------- /Scaler_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: Scaler.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf import descriptor as _descriptor 7 | from google.protobuf import message as _message 8 | from google.protobuf import reflection as _reflection 9 | from google.protobuf import symbol_database as _symbol_database 10 | from google.protobuf import descriptor_pb2 11 | # @@protoc_insertion_point(imports) 12 | 13 | _sym_db = _symbol_database.Default() 14 | 15 | 16 | 17 | 18 | DESCRIPTOR = _descriptor.FileDescriptor( 19 | name='Scaler.proto', 20 | package='CoreML.Specification', 21 | syntax='proto3', 22 | serialized_pb=_b('\n\x0cScaler.proto\x12\x14\x43oreML.Specification\"0\n\x06Scaler\x12\x12\n\nshiftValue\x18\x01 \x03(\x01\x12\x12\n\nscaleValue\x18\x02 \x03(\x01\x42\x02H\x03\x62\x06proto3') 23 | ) 24 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 25 | 26 | 27 | 28 | 29 | _SCALER = _descriptor.Descriptor( 30 | name='Scaler', 31 | full_name='CoreML.Specification.Scaler', 32 | filename=None, 33 | file=DESCRIPTOR, 34 | containing_type=None, 35 | fields=[ 36 | _descriptor.FieldDescriptor( 37 | name='shiftValue', full_name='CoreML.Specification.Scaler.shiftValue', index=0, 38 | number=1, type=1, cpp_type=5, label=3, 39 | has_default_value=False, default_value=[], 40 | message_type=None, enum_type=None, containing_type=None, 41 | is_extension=False, extension_scope=None, 42 | options=None), 43 | _descriptor.FieldDescriptor( 44 | name='scaleValue', full_name='CoreML.Specification.Scaler.scaleValue', index=1, 45 | number=2, type=1, cpp_type=5, label=3, 46 | has_default_value=False, default_value=[], 47 | message_type=None, enum_type=None, containing_type=None, 48 | is_extension=False, extension_scope=None, 49 | options=None), 50 | ], 51 | extensions=[ 52 | ], 53 | nested_types=[], 54 | enum_types=[ 55 | ], 56 | options=None, 57 | is_extendable=False, 58 | syntax='proto3', 59 | extension_ranges=[], 60 | oneofs=[ 61 | ], 62 | serialized_start=38, 63 | serialized_end=86, 64 | ) 65 | 66 | DESCRIPTOR.message_types_by_name['Scaler'] = _SCALER 67 | 68 | Scaler = _reflection.GeneratedProtocolMessageType('Scaler', (_message.Message,), dict( 69 | DESCRIPTOR = _SCALER, 70 | __module__ = 'Scaler_pb2' 71 | # @@protoc_insertion_point(class_scope:CoreML.Specification.Scaler) 72 | )) 73 | _sym_db.RegisterMessage(Scaler) 74 | 75 | 76 | DESCRIPTOR.has_options = True 77 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 78 | # @@protoc_insertion_point(module_scope) 79 | -------------------------------------------------------------------------------- /TreeEnsemble_pb2.py: -------------------------------------------------------------------------------- 1 | # Generated by the protocol buffer compiler. DO NOT EDIT! 2 | # source: TreeEnsemble.proto 3 | 4 | import sys 5 | _b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) 6 | from google.protobuf.internal import enum_type_wrapper 7 | from google.protobuf import descriptor as _descriptor 8 | from google.protobuf import message as _message 9 | from google.protobuf import reflection as _reflection 10 | from google.protobuf import symbol_database as _symbol_database 11 | from google.protobuf import descriptor_pb2 12 | # @@protoc_insertion_point(imports) 13 | 14 | _sym_db = _symbol_database.Default() 15 | 16 | 17 | import DataStructures_pb2 as DataStructures__pb2 18 | FeatureTypes__pb2 = DataStructures__pb2.FeatureTypes__pb2 19 | 20 | from DataStructures_pb2 import * 21 | 22 | DESCRIPTOR = _descriptor.FileDescriptor( 23 | name='TreeEnsemble.proto', 24 | package='CoreML.Specification', 25 | syntax='proto3', 26 | serialized_pb=_b('\n\x12TreeEnsemble.proto\x12\x14\x43oreML.Specification\x1a\x14\x44\x61taStructures.proto\"\xc4\x06\n\x16TreeEnsembleParameters\x12\x44\n\x05nodes\x18\x01 \x03(\x0b\x32\x35.CoreML.Specification.TreeEnsembleParameters.TreeNode\x12\x1f\n\x17numPredictionDimensions\x18\x02 \x01(\x04\x12\x1b\n\x13\x62\x61sePredictionValue\x18\x03 \x03(\x01\x1a\xa5\x05\n\x08TreeNode\x12\x0e\n\x06treeId\x18\x01 \x01(\x04\x12\x0e\n\x06nodeId\x18\x02 \x01(\x04\x12\\\n\x0cnodeBehavior\x18\x03 \x01(\x0e\x32\x46.CoreML.Specification.TreeEnsembleParameters.TreeNode.TreeNodeBehavior\x12\x1a\n\x12\x62ranchFeatureIndex\x18\n \x01(\x04\x12\x1a\n\x12\x62ranchFeatureValue\x18\x0b \x01(\x01\x12\x17\n\x0ftrueChildNodeId\x18\x0c \x01(\x04\x12\x18\n\x10\x66\x61lseChildNodeId\x18\r \x01(\x04\x12#\n\x1bmissingValueTracksTrueChild\x18\x0e \x01(\x08\x12\\\n\x0e\x65valuationInfo\x18\x14 \x03(\x0b\x32\x44.CoreML.Specification.TreeEnsembleParameters.TreeNode.EvaluationInfo\x12\x17\n\x0frelativeHitRate\x18\x1e \x01(\x01\x1a\x42\n\x0e\x45valuationInfo\x12\x17\n\x0f\x65valuationIndex\x18\x01 \x01(\x04\x12\x17\n\x0f\x65valuationValue\x18\x02 \x01(\x01\"\xcf\x01\n\x10TreeNodeBehavior\x12\x1e\n\x1a\x42ranchOnValueLessThanEqual\x10\x00\x12\x19\n\x15\x42ranchOnValueLessThan\x10\x01\x12!\n\x1d\x42ranchOnValueGreaterThanEqual\x10\x02\x12\x1c\n\x18\x42ranchOnValueGreaterThan\x10\x03\x12\x16\n\x12\x42ranchOnValueEqual\x10\x04\x12\x19\n\x15\x42ranchOnValueNotEqual\x10\x05\x12\x0c\n\x08LeafNode\x10\x06\"\xc7\x02\n\x16TreeEnsembleClassifier\x12\x42\n\x0ctreeEnsemble\x18\x01 \x01(\x0b\x32,.CoreML.Specification.TreeEnsembleParameters\x12Z\n\x17postEvaluationTransform\x18\x02 \x01(\x0e\x32\x39.CoreML.Specification.TreeEnsemblePostEvaluationTransform\x12?\n\x11stringClassLabels\x18\x64 \x01(\x0b\x32\".CoreML.Specification.StringVectorH\x00\x12=\n\x10int64ClassLabels\x18\x65 \x01(\x0b\x32!.CoreML.Specification.Int64VectorH\x00\x42\r\n\x0b\x43lassLabels\"\xb7\x01\n\x15TreeEnsembleRegressor\x12\x42\n\x0ctreeEnsemble\x18\x01 \x01(\x0b\x32,.CoreML.Specification.TreeEnsembleParameters\x12Z\n\x17postEvaluationTransform\x18\x02 \x01(\x0e\x32\x39.CoreML.Specification.TreeEnsemblePostEvaluationTransform*\x9d\x01\n#TreeEnsemblePostEvaluationTransform\x12\x0f\n\x0bNoTransform\x10\x00\x12\x1a\n\x16\x43lassification_SoftMax\x10\x01\x12\x17\n\x13Regression_Logistic\x10\x02\x12\x30\n,Classification_SoftMaxWithZeroClassReference\x10\x03\x42\x02H\x03P\x00\x62\x06proto3') 27 | , 28 | dependencies=[DataStructures__pb2.DESCRIPTOR,], 29 | public_dependencies=[DataStructures__pb2.DESCRIPTOR,]) 30 | _sym_db.RegisterFileDescriptor(DESCRIPTOR) 31 | 32 | _TREEENSEMBLEPOSTEVALUATIONTRANSFORM = _descriptor.EnumDescriptor( 33 | name='TreeEnsemblePostEvaluationTransform', 34 | full_name='CoreML.Specification.TreeEnsemblePostEvaluationTransform', 35 | filename=None, 36 | file=DESCRIPTOR, 37 | values=[ 38 | _descriptor.EnumValueDescriptor( 39 | name='NoTransform', index=0, number=0, 40 | options=None, 41 | type=None), 42 | _descriptor.EnumValueDescriptor( 43 | name='Classification_SoftMax', index=1, number=1, 44 | options=None, 45 | type=None), 46 | _descriptor.EnumValueDescriptor( 47 | name='Regression_Logistic', index=2, number=2, 48 | options=None, 49 | type=None), 50 | _descriptor.EnumValueDescriptor( 51 | name='Classification_SoftMaxWithZeroClassReference', index=3, number=3, 52 | options=None, 53 | type=None), 54 | ], 55 | containing_type=None, 56 | options=None, 57 | serialized_start=1422, 58 | serialized_end=1579, 59 | ) 60 | _sym_db.RegisterEnumDescriptor(_TREEENSEMBLEPOSTEVALUATIONTRANSFORM) 61 | 62 | TreeEnsemblePostEvaluationTransform = enum_type_wrapper.EnumTypeWrapper(_TREEENSEMBLEPOSTEVALUATIONTRANSFORM) 63 | NoTransform = 0 64 | Classification_SoftMax = 1 65 | Regression_Logistic = 2 66 | Classification_SoftMaxWithZeroClassReference = 3 67 | 68 | 69 | _TREEENSEMBLEPARAMETERS_TREENODE_TREENODEBEHAVIOR = _descriptor.EnumDescriptor( 70 | name='TreeNodeBehavior', 71 | full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.TreeNodeBehavior', 72 | filename=None, 73 | file=DESCRIPTOR, 74 | values=[ 75 | _descriptor.EnumValueDescriptor( 76 | name='BranchOnValueLessThanEqual', index=0, number=0, 77 | options=None, 78 | type=None), 79 | _descriptor.EnumValueDescriptor( 80 | name='BranchOnValueLessThan', index=1, number=1, 81 | options=None, 82 | type=None), 83 | _descriptor.EnumValueDescriptor( 84 | name='BranchOnValueGreaterThanEqual', index=2, number=2, 85 | options=None, 86 | type=None), 87 | _descriptor.EnumValueDescriptor( 88 | name='BranchOnValueGreaterThan', index=3, number=3, 89 | options=None, 90 | type=None), 91 | _descriptor.EnumValueDescriptor( 92 | name='BranchOnValueEqual', index=4, number=4, 93 | options=None, 94 | type=None), 95 | _descriptor.EnumValueDescriptor( 96 | name='BranchOnValueNotEqual', index=5, number=5, 97 | options=None, 98 | type=None), 99 | _descriptor.EnumValueDescriptor( 100 | name='LeafNode', index=6, number=6, 101 | options=None, 102 | type=None), 103 | ], 104 | containing_type=None, 105 | options=None, 106 | serialized_start=696, 107 | serialized_end=903, 108 | ) 109 | _sym_db.RegisterEnumDescriptor(_TREEENSEMBLEPARAMETERS_TREENODE_TREENODEBEHAVIOR) 110 | 111 | 112 | _TREEENSEMBLEPARAMETERS_TREENODE_EVALUATIONINFO = _descriptor.Descriptor( 113 | name='EvaluationInfo', 114 | full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.EvaluationInfo', 115 | filename=None, 116 | file=DESCRIPTOR, 117 | containing_type=None, 118 | fields=[ 119 | _descriptor.FieldDescriptor( 120 | name='evaluationIndex', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.EvaluationInfo.evaluationIndex', index=0, 121 | number=1, type=4, cpp_type=4, label=1, 122 | has_default_value=False, default_value=0, 123 | message_type=None, enum_type=None, containing_type=None, 124 | is_extension=False, extension_scope=None, 125 | options=None), 126 | _descriptor.FieldDescriptor( 127 | name='evaluationValue', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.EvaluationInfo.evaluationValue', index=1, 128 | number=2, type=1, cpp_type=5, label=1, 129 | has_default_value=False, default_value=float(0), 130 | message_type=None, enum_type=None, containing_type=None, 131 | is_extension=False, extension_scope=None, 132 | options=None), 133 | ], 134 | extensions=[ 135 | ], 136 | nested_types=[], 137 | enum_types=[ 138 | ], 139 | options=None, 140 | is_extendable=False, 141 | syntax='proto3', 142 | extension_ranges=[], 143 | oneofs=[ 144 | ], 145 | serialized_start=627, 146 | serialized_end=693, 147 | ) 148 | 149 | _TREEENSEMBLEPARAMETERS_TREENODE = _descriptor.Descriptor( 150 | name='TreeNode', 151 | full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode', 152 | filename=None, 153 | file=DESCRIPTOR, 154 | containing_type=None, 155 | fields=[ 156 | _descriptor.FieldDescriptor( 157 | name='treeId', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.treeId', index=0, 158 | number=1, type=4, cpp_type=4, label=1, 159 | has_default_value=False, default_value=0, 160 | message_type=None, enum_type=None, containing_type=None, 161 | is_extension=False, extension_scope=None, 162 | options=None), 163 | _descriptor.FieldDescriptor( 164 | name='nodeId', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.nodeId', index=1, 165 | number=2, type=4, cpp_type=4, label=1, 166 | has_default_value=False, default_value=0, 167 | message_type=None, enum_type=None, containing_type=None, 168 | is_extension=False, extension_scope=None, 169 | options=None), 170 | _descriptor.FieldDescriptor( 171 | name='nodeBehavior', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.nodeBehavior', index=2, 172 | number=3, type=14, cpp_type=8, label=1, 173 | has_default_value=False, default_value=0, 174 | message_type=None, enum_type=None, containing_type=None, 175 | is_extension=False, extension_scope=None, 176 | options=None), 177 | _descriptor.FieldDescriptor( 178 | name='branchFeatureIndex', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.branchFeatureIndex', index=3, 179 | number=10, type=4, cpp_type=4, label=1, 180 | has_default_value=False, default_value=0, 181 | message_type=None, enum_type=None, containing_type=None, 182 | is_extension=False, extension_scope=None, 183 | options=None), 184 | _descriptor.FieldDescriptor( 185 | name='branchFeatureValue', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.branchFeatureValue', index=4, 186 | number=11, type=1, cpp_type=5, label=1, 187 | has_default_value=False, default_value=float(0), 188 | message_type=None, enum_type=None, containing_type=None, 189 | is_extension=False, extension_scope=None, 190 | options=None), 191 | _descriptor.FieldDescriptor( 192 | name='trueChildNodeId', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.trueChildNodeId', index=5, 193 | number=12, type=4, cpp_type=4, label=1, 194 | has_default_value=False, default_value=0, 195 | message_type=None, enum_type=None, containing_type=None, 196 | is_extension=False, extension_scope=None, 197 | options=None), 198 | _descriptor.FieldDescriptor( 199 | name='falseChildNodeId', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.falseChildNodeId', index=6, 200 | number=13, type=4, cpp_type=4, label=1, 201 | has_default_value=False, default_value=0, 202 | message_type=None, enum_type=None, containing_type=None, 203 | is_extension=False, extension_scope=None, 204 | options=None), 205 | _descriptor.FieldDescriptor( 206 | name='missingValueTracksTrueChild', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.missingValueTracksTrueChild', index=7, 207 | number=14, type=8, cpp_type=7, label=1, 208 | has_default_value=False, default_value=False, 209 | message_type=None, enum_type=None, containing_type=None, 210 | is_extension=False, extension_scope=None, 211 | options=None), 212 | _descriptor.FieldDescriptor( 213 | name='evaluationInfo', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.evaluationInfo', index=8, 214 | number=20, type=11, cpp_type=10, label=3, 215 | has_default_value=False, default_value=[], 216 | message_type=None, enum_type=None, containing_type=None, 217 | is_extension=False, extension_scope=None, 218 | options=None), 219 | _descriptor.FieldDescriptor( 220 | name='relativeHitRate', full_name='CoreML.Specification.TreeEnsembleParameters.TreeNode.relativeHitRate', index=9, 221 | number=30, type=1, cpp_type=5, label=1, 222 | has_default_value=False, default_value=float(0), 223 | message_type=None, enum_type=None, containing_type=None, 224 | is_extension=False, extension_scope=None, 225 | options=None), 226 | ], 227 | extensions=[ 228 | ], 229 | nested_types=[_TREEENSEMBLEPARAMETERS_TREENODE_EVALUATIONINFO, ], 230 | enum_types=[ 231 | _TREEENSEMBLEPARAMETERS_TREENODE_TREENODEBEHAVIOR, 232 | ], 233 | options=None, 234 | is_extendable=False, 235 | syntax='proto3', 236 | extension_ranges=[], 237 | oneofs=[ 238 | ], 239 | serialized_start=226, 240 | serialized_end=903, 241 | ) 242 | 243 | _TREEENSEMBLEPARAMETERS = _descriptor.Descriptor( 244 | name='TreeEnsembleParameters', 245 | full_name='CoreML.Specification.TreeEnsembleParameters', 246 | filename=None, 247 | file=DESCRIPTOR, 248 | containing_type=None, 249 | fields=[ 250 | _descriptor.FieldDescriptor( 251 | name='nodes', full_name='CoreML.Specification.TreeEnsembleParameters.nodes', index=0, 252 | number=1, type=11, cpp_type=10, label=3, 253 | has_default_value=False, default_value=[], 254 | message_type=None, enum_type=None, containing_type=None, 255 | is_extension=False, extension_scope=None, 256 | options=None), 257 | _descriptor.FieldDescriptor( 258 | name='numPredictionDimensions', full_name='CoreML.Specification.TreeEnsembleParameters.numPredictionDimensions', index=1, 259 | number=2, type=4, cpp_type=4, label=1, 260 | has_default_value=False, default_value=0, 261 | message_type=None, enum_type=None, containing_type=None, 262 | is_extension=False, extension_scope=None, 263 | options=None), 264 | _descriptor.FieldDescriptor( 265 | name='basePredictionValue', full_name='CoreML.Specification.TreeEnsembleParameters.basePredictionValue', index=2, 266 | number=3, type=1, cpp_type=5, label=3, 267 | has_default_value=False, default_value=[], 268 | message_type=None, enum_type=None, containing_type=None, 269 | is_extension=False, extension_scope=None, 270 | options=None), 271 | ], 272 | extensions=[ 273 | ], 274 | nested_types=[_TREEENSEMBLEPARAMETERS_TREENODE, ], 275 | enum_types=[ 276 | ], 277 | options=None, 278 | is_extendable=False, 279 | syntax='proto3', 280 | extension_ranges=[], 281 | oneofs=[ 282 | ], 283 | serialized_start=67, 284 | serialized_end=903, 285 | ) 286 | 287 | 288 | _TREEENSEMBLECLASSIFIER = _descriptor.Descriptor( 289 | name='TreeEnsembleClassifier', 290 | full_name='CoreML.Specification.TreeEnsembleClassifier', 291 | filename=None, 292 | file=DESCRIPTOR, 293 | containing_type=None, 294 | fields=[ 295 | _descriptor.FieldDescriptor( 296 | name='treeEnsemble', full_name='CoreML.Specification.TreeEnsembleClassifier.treeEnsemble', index=0, 297 | number=1, type=11, cpp_type=10, label=1, 298 | has_default_value=False, default_value=None, 299 | message_type=None, enum_type=None, containing_type=None, 300 | is_extension=False, extension_scope=None, 301 | options=None), 302 | _descriptor.FieldDescriptor( 303 | name='postEvaluationTransform', full_name='CoreML.Specification.TreeEnsembleClassifier.postEvaluationTransform', index=1, 304 | number=2, type=14, cpp_type=8, label=1, 305 | has_default_value=False, default_value=0, 306 | message_type=None, enum_type=None, containing_type=None, 307 | is_extension=False, extension_scope=None, 308 | options=None), 309 | _descriptor.FieldDescriptor( 310 | name='stringClassLabels', full_name='CoreML.Specification.TreeEnsembleClassifier.stringClassLabels', index=2, 311 | number=100, type=11, cpp_type=10, label=1, 312 | has_default_value=False, default_value=None, 313 | message_type=None, enum_type=None, containing_type=None, 314 | is_extension=False, extension_scope=None, 315 | options=None), 316 | _descriptor.FieldDescriptor( 317 | name='int64ClassLabels', full_name='CoreML.Specification.TreeEnsembleClassifier.int64ClassLabels', index=3, 318 | number=101, type=11, cpp_type=10, label=1, 319 | has_default_value=False, default_value=None, 320 | message_type=None, enum_type=None, containing_type=None, 321 | is_extension=False, extension_scope=None, 322 | options=None), 323 | ], 324 | extensions=[ 325 | ], 326 | nested_types=[], 327 | enum_types=[ 328 | ], 329 | options=None, 330 | is_extendable=False, 331 | syntax='proto3', 332 | extension_ranges=[], 333 | oneofs=[ 334 | _descriptor.OneofDescriptor( 335 | name='ClassLabels', full_name='CoreML.Specification.TreeEnsembleClassifier.ClassLabels', 336 | index=0, containing_type=None, fields=[]), 337 | ], 338 | serialized_start=906, 339 | serialized_end=1233, 340 | ) 341 | 342 | 343 | _TREEENSEMBLEREGRESSOR = _descriptor.Descriptor( 344 | name='TreeEnsembleRegressor', 345 | full_name='CoreML.Specification.TreeEnsembleRegressor', 346 | filename=None, 347 | file=DESCRIPTOR, 348 | containing_type=None, 349 | fields=[ 350 | _descriptor.FieldDescriptor( 351 | name='treeEnsemble', full_name='CoreML.Specification.TreeEnsembleRegressor.treeEnsemble', index=0, 352 | number=1, type=11, cpp_type=10, label=1, 353 | has_default_value=False, default_value=None, 354 | message_type=None, enum_type=None, containing_type=None, 355 | is_extension=False, extension_scope=None, 356 | options=None), 357 | _descriptor.FieldDescriptor( 358 | name='postEvaluationTransform', full_name='CoreML.Specification.TreeEnsembleRegressor.postEvaluationTransform', index=1, 359 | number=2, type=14, cpp_type=8, label=1, 360 | has_default_value=False, default_value=0, 361 | message_type=None, enum_type=None, containing_type=None, 362 | is_extension=False, extension_scope=None, 363 | options=None), 364 | ], 365 | extensions=[ 366 | ], 367 | nested_types=[], 368 | enum_types=[ 369 | ], 370 | options=None, 371 | is_extendable=False, 372 | syntax='proto3', 373 | extension_ranges=[], 374 | oneofs=[ 375 | ], 376 | serialized_start=1236, 377 | serialized_end=1419, 378 | ) 379 | 380 | _TREEENSEMBLEPARAMETERS_TREENODE_EVALUATIONINFO.containing_type = _TREEENSEMBLEPARAMETERS_TREENODE 381 | _TREEENSEMBLEPARAMETERS_TREENODE.fields_by_name['nodeBehavior'].enum_type = _TREEENSEMBLEPARAMETERS_TREENODE_TREENODEBEHAVIOR 382 | _TREEENSEMBLEPARAMETERS_TREENODE.fields_by_name['evaluationInfo'].message_type = _TREEENSEMBLEPARAMETERS_TREENODE_EVALUATIONINFO 383 | _TREEENSEMBLEPARAMETERS_TREENODE.containing_type = _TREEENSEMBLEPARAMETERS 384 | _TREEENSEMBLEPARAMETERS_TREENODE_TREENODEBEHAVIOR.containing_type = _TREEENSEMBLEPARAMETERS_TREENODE 385 | _TREEENSEMBLEPARAMETERS.fields_by_name['nodes'].message_type = _TREEENSEMBLEPARAMETERS_TREENODE 386 | _TREEENSEMBLECLASSIFIER.fields_by_name['treeEnsemble'].message_type = _TREEENSEMBLEPARAMETERS 387 | _TREEENSEMBLECLASSIFIER.fields_by_name['postEvaluationTransform'].enum_type = _TREEENSEMBLEPOSTEVALUATIONTRANSFORM 388 | _TREEENSEMBLECLASSIFIER.fields_by_name['stringClassLabels'].message_type = DataStructures__pb2._STRINGVECTOR 389 | _TREEENSEMBLECLASSIFIER.fields_by_name['int64ClassLabels'].message_type = DataStructures__pb2._INT64VECTOR 390 | _TREEENSEMBLECLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 391 | _TREEENSEMBLECLASSIFIER.fields_by_name['stringClassLabels']) 392 | _TREEENSEMBLECLASSIFIER.fields_by_name['stringClassLabels'].containing_oneof = _TREEENSEMBLECLASSIFIER.oneofs_by_name['ClassLabels'] 393 | _TREEENSEMBLECLASSIFIER.oneofs_by_name['ClassLabels'].fields.append( 394 | _TREEENSEMBLECLASSIFIER.fields_by_name['int64ClassLabels']) 395 | _TREEENSEMBLECLASSIFIER.fields_by_name['int64ClassLabels'].containing_oneof = _TREEENSEMBLECLASSIFIER.oneofs_by_name['ClassLabels'] 396 | _TREEENSEMBLEREGRESSOR.fields_by_name['treeEnsemble'].message_type = _TREEENSEMBLEPARAMETERS 397 | _TREEENSEMBLEREGRESSOR.fields_by_name['postEvaluationTransform'].enum_type = _TREEENSEMBLEPOSTEVALUATIONTRANSFORM 398 | DESCRIPTOR.message_types_by_name['TreeEnsembleParameters'] = _TREEENSEMBLEPARAMETERS 399 | DESCRIPTOR.message_types_by_name['TreeEnsembleClassifier'] = _TREEENSEMBLECLASSIFIER 400 | DESCRIPTOR.message_types_by_name['TreeEnsembleRegressor'] = _TREEENSEMBLEREGRESSOR 401 | DESCRIPTOR.enum_types_by_name['TreeEnsemblePostEvaluationTransform'] = _TREEENSEMBLEPOSTEVALUATIONTRANSFORM 402 | 403 | TreeEnsembleParameters = _reflection.GeneratedProtocolMessageType('TreeEnsembleParameters', (_message.Message,), dict( 404 | 405 | TreeNode = _reflection.GeneratedProtocolMessageType('TreeNode', (_message.Message,), dict( 406 | 407 | EvaluationInfo = _reflection.GeneratedProtocolMessageType('EvaluationInfo', (_message.Message,), dict( 408 | DESCRIPTOR = _TREEENSEMBLEPARAMETERS_TREENODE_EVALUATIONINFO, 409 | __module__ = 'TreeEnsemble_pb2' 410 | # @@protoc_insertion_point(class_scope:CoreML.Specification.TreeEnsembleParameters.TreeNode.EvaluationInfo) 411 | )) 412 | , 413 | DESCRIPTOR = _TREEENSEMBLEPARAMETERS_TREENODE, 414 | __module__ = 'TreeEnsemble_pb2' 415 | # @@protoc_insertion_point(class_scope:CoreML.Specification.TreeEnsembleParameters.TreeNode) 416 | )) 417 | , 418 | DESCRIPTOR = _TREEENSEMBLEPARAMETERS, 419 | __module__ = 'TreeEnsemble_pb2' 420 | # @@protoc_insertion_point(class_scope:CoreML.Specification.TreeEnsembleParameters) 421 | )) 422 | _sym_db.RegisterMessage(TreeEnsembleParameters) 423 | _sym_db.RegisterMessage(TreeEnsembleParameters.TreeNode) 424 | _sym_db.RegisterMessage(TreeEnsembleParameters.TreeNode.EvaluationInfo) 425 | 426 | TreeEnsembleClassifier = _reflection.GeneratedProtocolMessageType('TreeEnsembleClassifier', (_message.Message,), dict( 427 | DESCRIPTOR = _TREEENSEMBLECLASSIFIER, 428 | __module__ = 'TreeEnsemble_pb2' 429 | # @@protoc_insertion_point(class_scope:CoreML.Specification.TreeEnsembleClassifier) 430 | )) 431 | _sym_db.RegisterMessage(TreeEnsembleClassifier) 432 | 433 | TreeEnsembleRegressor = _reflection.GeneratedProtocolMessageType('TreeEnsembleRegressor', (_message.Message,), dict( 434 | DESCRIPTOR = _TREEENSEMBLEREGRESSOR, 435 | __module__ = 'TreeEnsemble_pb2' 436 | # @@protoc_insertion_point(class_scope:CoreML.Specification.TreeEnsembleRegressor) 437 | )) 438 | _sym_db.RegisterMessage(TreeEnsembleRegressor) 439 | 440 | 441 | DESCRIPTOR.has_options = True 442 | DESCRIPTOR._options = _descriptor._ParseOptions(descriptor_pb2.FileOptions(), _b('H\003')) 443 | # @@protoc_insertion_point(module_scope) 444 | -------------------------------------------------------------------------------- /lookieloo.py: -------------------------------------------------------------------------------- 1 | import os 2 | import sys 3 | import Model_pb2 4 | 5 | args = sys.argv[1:] 6 | if len(args) != 1: 7 | print("usage: %s name.mlmodel" % os.path.basename(__file__)) 8 | exit(-1) 9 | model_path = args[0] 10 | 11 | model = Model_pb2.Model() 12 | 13 | print("Loading model...") 14 | f = open(model_path, "rb") 15 | model.ParseFromString(f.read()) 16 | f.close() 17 | 18 | print(model.description) 19 | 20 | print("\nLayers:") 21 | for layer in model.neuralNetworkClassifier.layers: 22 | print(layer.name) 23 | if layer.HasField("convolution"): 24 | print("\tis convolution") 25 | print("\tkernel size:", layer.convolution.kernelSize) 26 | print("\tinput channels", layer.convolution.kernelChannels) 27 | print("\toutput channels", layer.convolution.outputChannels) 28 | 29 | print("\nPreprocessing:") 30 | print(model.neuralNetworkClassifier.preprocessing) 31 | 32 | # Enable this to see the whole thing. Warning: it will also print all the weights! 33 | #print(model) 34 | --------------------------------------------------------------------------------