google-research

1
# coding=utf-8
2
# Copyright 2024 The Google Research Authors.
3
#
4
# Licensed under the Apache License, Version 2.0 (the "License");
5
# you may not use this file except in compliance with the License.
6
# You may obtain a copy of the License at
7
#
8
#     http://www.apache.org/licenses/LICENSE-2.0
9
#
10
# Unless required by applicable law or agreed to in writing, software
11
# distributed under the License is distributed on an "AS IS" BASIS,
12
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
# See the License for the specific language governing permissions and
14
# limitations under the License.
15

16
# -*- coding: utf-8 -*-
17
# pylint: skip-file
18
# Generated by the protocol buffer compiler.  DO NOT EDIT!
19
# source: task.proto
20

21
from google.protobuf.internal import enum_type_wrapper
22
from google.protobuf import descriptor as _descriptor
23
from google.protobuf import message as _message
24
from google.protobuf import reflection as _reflection
25
from google.protobuf import symbol_database as _symbol_database
26
# @@protoc_insertion_point(imports)
27

28
_sym_db = _symbol_database.Default()
29

30

31

32

33
DESCRIPTOR = _descriptor.FileDescriptor(
34
  name='task.proto',
35
  package='automl_zero',
36
  syntax='proto2',
37
  serialized_options=None,
38
  serialized_pb=b'\n\ntask.proto\x12\x0b\x61utoml_zero\"6\n\x0eTaskCollection\x12$\n\x05tasks\x18\x01 \x03(\x0b\x32\x15.automl_zero.TaskSpec\"\xa7\x06\n\x08TaskSpec\x12\x15\n\rfeatures_size\x18\r \x01(\x05\x12\x1a\n\x12num_train_examples\x18\x01 \x01(\x05\x12\x1b\n\x10num_train_epochs\x18\x15 \x01(\x05:\x01\x31\x12\x1a\n\x12num_valid_examples\x18\x02 \x01(\x05\x12\x11\n\tnum_tasks\x18\x03 \x01(\x05\x12\x12\n\ndata_seeds\x18\x04 \x03(\r\x12\x13\n\x0bparam_seeds\x18\x05 \x03(\r\x12(\n\teval_type\x18\x1c \x01(\x0e\x32\x15.automl_zero.EvalType\x12T\n\x1dscalar_linear_regression_task\x18\x06 \x01(\x0b\x32+.automl_zero.ScalarLinearRegressionTaskSpecH\x00\x12Y\n scalar_2layer_nn_regression_task\x18\x07 \x01(\x0b\x32-.automl_zero.Scalar2LayerNNRegressionTaskSpecH\x00\x12^\n$projected_binary_classification_task\x18\x18 \x01(\x0b\x32..automl_zero.ProjectedBinaryClassificationTaskH\x00\x12>\n\x14unit_test_fixed_task\x18( \x01(\x0b\x32\x1e.automl_zero.UnitTestFixedTaskH\x00\x12\x42\n\x14unit_test_zeros_task\x18- \x01(\x0b\x32\".automl_zero.UnitTestZerosTaskSpecH\x00\x12@\n\x13unit_test_ones_task\x18. \x01(\x0b\x32!.automl_zero.UnitTestOnesTaskSpecH\x00\x12J\n\x18unit_test_increment_task\x18/ \x01(\x0b\x32&.automl_zero.UnitTestIncrementTaskSpecH\x00\x12\x19\n\x11num_test_examples\x18\x12 \x01(\x05\x42\x0b\n\ttask_type\" \n\x1eScalarLinearRegressionTaskSpec\"\"\n Scalar2LayerNNRegressionTaskSpec\"\xb5\x02\n!ProjectedBinaryClassificationTask\x12\x16\n\x0epositive_class\x18\x01 \x01(\x05\x12\x16\n\x0enegative_class\x18\x02 \x01(\x05\x12\x14\n\x0c\x64\x61taset_name\x18\x03 \x01(\t\x12\x0e\n\x04path\x18\x04 \x01(\tH\x00\x12\x32\n\x07\x64\x61taset\x18\x05 \x01(\x0b\x32\x1f.automl_zero.ScalarLabelDatasetH\x00\x12.\n\x0eheld_out_pairs\x18\x06 \x03(\x0b\x32\x16.automl_zero.ClassPair\x12\"\n\x17min_supported_data_seed\x18\x07 \x01(\x05:\x01\x30\x12#\n\x17max_supported_data_seed\x18\x08 \x01(\x05:\x02\x31\x30\x42\r\n\x0btask_source\";\n\tClassPair\x12\x16\n\x0epositive_class\x18\x01 \x01(\x05\x12\x16\n\x0enegative_class\x18\x02 \x01(\x05\"\xf0\x01\n\x12ScalarLabelDataset\x12\x32\n\x0etrain_features\x18\x01 \x03(\x0b\x32\x1a.automl_zero.FeatureVector\x12\x14\n\x0ctrain_labels\x18\x02 \x03(\x02\x12\x32\n\x0evalid_features\x18\x03 \x03(\x0b\x32\x1a.automl_zero.FeatureVector\x12\x14\n\x0cvalid_labels\x18\x04 \x03(\x02\x12\x31\n\rtest_features\x18\x05 \x03(\x0b\x32\x1a.automl_zero.FeatureVector\x12\x13\n\x0btest_labels\x18\x06 \x03(\x02\"!\n\rFeatureVector\x12\x10\n\x08\x66\x65\x61tures\x18\x01 \x03(\x02\"\x85\x03\n\x11UnitTestFixedTask\x12<\n\x0etrain_features\x18\x01 \x03(\x0b\x32$.automl_zero.UnitTestFixedTaskVector\x12:\n\x0ctrain_labels\x18\x02 \x03(\x0b\x32$.automl_zero.UnitTestFixedTaskVector\x12<\n\x0evalid_features\x18\x03 \x03(\x0b\x32$.automl_zero.UnitTestFixedTaskVector\x12:\n\x0cvalid_labels\x18\x04 \x03(\x0b\x32$.automl_zero.UnitTestFixedTaskVector\x12;\n\rtest_features\x18\x05 \x03(\x0b\x32$.automl_zero.UnitTestFixedTaskVector\x12\x39\n\x0btest_labels\x18\x06 \x03(\x0b\x32$.automl_zero.UnitTestFixedTaskVectorJ\x04\x08\x07\x10\x08\"+\n\x17UnitTestFixedTaskVector\x12\x10\n\x08\x65lements\x18\x01 \x03(\x01\"\x17\n\x15UnitTestZerosTaskSpec\"\x16\n\x14UnitTestOnesTaskSpec\"1\n\x19UnitTestIncrementTaskSpec\x12\x14\n\tincrement\x18\x01 \x01(\x01:\x01\x31*>\n\x08\x45valType\x12\x15\n\x11INVALID_EVAL_TYPE\x10\x00\x12\r\n\tRMS_ERROR\x10\x01\x12\x0c\n\x08\x41\x43\x43URACY\x10\x04*$\n\x0e\x41\x63tivationType\x12\x08\n\x04RELU\x10\x00\x12\x08\n\x04TANH\x10\x01'
39
)
40

41
_EVALTYPE = _descriptor.EnumDescriptor(
42
  name='EvalType',
43
  full_name='automl_zero.EvalType',
44
  filename=None,
45
  file=DESCRIPTOR,
46
  values=[
47
    _descriptor.EnumValueDescriptor(
48
      name='INVALID_EVAL_TYPE', index=0, number=0,
49
      serialized_options=None,
50
      type=None),
51
    _descriptor.EnumValueDescriptor(
52
      name='RMS_ERROR', index=1, number=1,
53
      serialized_options=None,
54
      type=None),
55
    _descriptor.EnumValueDescriptor(
56
      name='ACCURACY', index=2, number=4,
57
      serialized_options=None,
58
      type=None),
59
  ],
60
  containing_type=None,
61
  serialized_options=None,
62
  serialized_start=2151,
63
  serialized_end=2213,
64
)
65
_sym_db.RegisterEnumDescriptor(_EVALTYPE)
66

67
EvalType = enum_type_wrapper.EnumTypeWrapper(_EVALTYPE)
68
_ACTIVATIONTYPE = _descriptor.EnumDescriptor(
69
  name='ActivationType',
70
  full_name='automl_zero.ActivationType',
71
  filename=None,
72
  file=DESCRIPTOR,
73
  values=[
74
    _descriptor.EnumValueDescriptor(
75
      name='RELU', index=0, number=0,
76
      serialized_options=None,
77
      type=None),
78
    _descriptor.EnumValueDescriptor(
79
      name='TANH', index=1, number=1,
80
      serialized_options=None,
81
      type=None),
82
  ],
83
  containing_type=None,
84
  serialized_options=None,
85
  serialized_start=2215,
86
  serialized_end=2251,
87
)
88
_sym_db.RegisterEnumDescriptor(_ACTIVATIONTYPE)
89

90
ActivationType = enum_type_wrapper.EnumTypeWrapper(_ACTIVATIONTYPE)
91
INVALID_EVAL_TYPE = 0
92
RMS_ERROR = 1
93
ACCURACY = 4
94
RELU = 0
95
TANH = 1
96

97

98

99
_TASKCOLLECTION = _descriptor.Descriptor(
100
  name='TaskCollection',
101
  full_name='automl_zero.TaskCollection',
102
  filename=None,
103
  file=DESCRIPTOR,
104
  containing_type=None,
105
  fields=[
106
    _descriptor.FieldDescriptor(
107
      name='tasks', full_name='automl_zero.TaskCollection.tasks', index=0,
108
      number=1, type=11, cpp_type=10, label=3,
109
      has_default_value=False, default_value=[],
110
      message_type=None, enum_type=None, containing_type=None,
111
      is_extension=False, extension_scope=None,
112
      serialized_options=None, file=DESCRIPTOR),
113
  ],
114
  extensions=[
115
  ],
116
  nested_types=[],
117
  enum_types=[
118
  ],
119
  serialized_options=None,
120
  is_extendable=False,
121
  syntax='proto2',
122
  extension_ranges=[],
123
  oneofs=[
124
  ],
125
  serialized_start=27,
126
  serialized_end=81,
127
)
128

129

130
_TASKSPEC = _descriptor.Descriptor(
131
  name='TaskSpec',
132
  full_name='automl_zero.TaskSpec',
133
  filename=None,
134
  file=DESCRIPTOR,
135
  containing_type=None,
136
  fields=[
137
    _descriptor.FieldDescriptor(
138
      name='features_size', full_name='automl_zero.TaskSpec.features_size', index=0,
139
      number=13, type=5, cpp_type=1, label=1,
140
      has_default_value=False, default_value=0,
141
      message_type=None, enum_type=None, containing_type=None,
142
      is_extension=False, extension_scope=None,
143
      serialized_options=None, file=DESCRIPTOR),
144
    _descriptor.FieldDescriptor(
145
      name='num_train_examples', full_name='automl_zero.TaskSpec.num_train_examples', index=1,
146
      number=1, type=5, cpp_type=1, label=1,
147
      has_default_value=False, default_value=0,
148
      message_type=None, enum_type=None, containing_type=None,
149
      is_extension=False, extension_scope=None,
150
      serialized_options=None, file=DESCRIPTOR),
151
    _descriptor.FieldDescriptor(
152
      name='num_train_epochs', full_name='automl_zero.TaskSpec.num_train_epochs', index=2,
153
      number=21, type=5, cpp_type=1, label=1,
154
      has_default_value=True, default_value=1,
155
      message_type=None, enum_type=None, containing_type=None,
156
      is_extension=False, extension_scope=None,
157
      serialized_options=None, file=DESCRIPTOR),
158
    _descriptor.FieldDescriptor(
159
      name='num_valid_examples', full_name='automl_zero.TaskSpec.num_valid_examples', index=3,
160
      number=2, type=5, cpp_type=1, label=1,
161
      has_default_value=False, default_value=0,
162
      message_type=None, enum_type=None, containing_type=None,
163
      is_extension=False, extension_scope=None,
164
      serialized_options=None, file=DESCRIPTOR),
165
    _descriptor.FieldDescriptor(
166
      name='num_tasks', full_name='automl_zero.TaskSpec.num_tasks', index=4,
167
      number=3, type=5, cpp_type=1, label=1,
168
      has_default_value=False, default_value=0,
169
      message_type=None, enum_type=None, containing_type=None,
170
      is_extension=False, extension_scope=None,
171
      serialized_options=None, file=DESCRIPTOR),
172
    _descriptor.FieldDescriptor(
173
      name='data_seeds', full_name='automl_zero.TaskSpec.data_seeds', index=5,
174
      number=4, type=13, cpp_type=3, label=3,
175
      has_default_value=False, default_value=[],
176
      message_type=None, enum_type=None, containing_type=None,
177
      is_extension=False, extension_scope=None,
178
      serialized_options=None, file=DESCRIPTOR),
179
    _descriptor.FieldDescriptor(
180
      name='param_seeds', full_name='automl_zero.TaskSpec.param_seeds', index=6,
181
      number=5, type=13, cpp_type=3, label=3,
182
      has_default_value=False, default_value=[],
183
      message_type=None, enum_type=None, containing_type=None,
184
      is_extension=False, extension_scope=None,
185
      serialized_options=None, file=DESCRIPTOR),
186
    _descriptor.FieldDescriptor(
187
      name='eval_type', full_name='automl_zero.TaskSpec.eval_type', index=7,
188
      number=28, type=14, cpp_type=8, label=1,
189
      has_default_value=False, default_value=0,
190
      message_type=None, enum_type=None, containing_type=None,
191
      is_extension=False, extension_scope=None,
192
      serialized_options=None, file=DESCRIPTOR),
193
    _descriptor.FieldDescriptor(
194
      name='scalar_linear_regression_task', full_name='automl_zero.TaskSpec.scalar_linear_regression_task', index=8,
195
      number=6, 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
      serialized_options=None, file=DESCRIPTOR),
200
    _descriptor.FieldDescriptor(
201
      name='scalar_2layer_nn_regression_task', full_name='automl_zero.TaskSpec.scalar_2layer_nn_regression_task', index=9,
202
      number=7, 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
      serialized_options=None, file=DESCRIPTOR),
207
    _descriptor.FieldDescriptor(
208
      name='projected_binary_classification_task', full_name='automl_zero.TaskSpec.projected_binary_classification_task', index=10,
209
      number=24, type=11, cpp_type=10, label=1,
210
      has_default_value=False, default_value=None,
211
      message_type=None, enum_type=None, containing_type=None,
212
      is_extension=False, extension_scope=None,
213
      serialized_options=None, file=DESCRIPTOR),
214
    _descriptor.FieldDescriptor(
215
      name='unit_test_fixed_task', full_name='automl_zero.TaskSpec.unit_test_fixed_task', index=11,
216
      number=40, type=11, cpp_type=10, label=1,
217
      has_default_value=False, default_value=None,
218
      message_type=None, enum_type=None, containing_type=None,
219
      is_extension=False, extension_scope=None,
220
      serialized_options=None, file=DESCRIPTOR),
221
    _descriptor.FieldDescriptor(
222
      name='unit_test_zeros_task', full_name='automl_zero.TaskSpec.unit_test_zeros_task', index=12,
223
      number=45, type=11, cpp_type=10, label=1,
224
      has_default_value=False, default_value=None,
225
      message_type=None, enum_type=None, containing_type=None,
226
      is_extension=False, extension_scope=None,
227
      serialized_options=None, file=DESCRIPTOR),
228
    _descriptor.FieldDescriptor(
229
      name='unit_test_ones_task', full_name='automl_zero.TaskSpec.unit_test_ones_task', index=13,
230
      number=46, type=11, cpp_type=10, label=1,
231
      has_default_value=False, default_value=None,
232
      message_type=None, enum_type=None, containing_type=None,
233
      is_extension=False, extension_scope=None,
234
      serialized_options=None, file=DESCRIPTOR),
235
    _descriptor.FieldDescriptor(
236
      name='unit_test_increment_task', full_name='automl_zero.TaskSpec.unit_test_increment_task', index=14,
237
      number=47, type=11, cpp_type=10, label=1,
238
      has_default_value=False, default_value=None,
239
      message_type=None, enum_type=None, containing_type=None,
240
      is_extension=False, extension_scope=None,
241
      serialized_options=None, file=DESCRIPTOR),
242
    _descriptor.FieldDescriptor(
243
      name='num_test_examples', full_name='automl_zero.TaskSpec.num_test_examples', index=15,
244
      number=18, type=5, cpp_type=1, label=1,
245
      has_default_value=False, default_value=0,
246
      message_type=None, enum_type=None, containing_type=None,
247
      is_extension=False, extension_scope=None,
248
      serialized_options=None, file=DESCRIPTOR),
249
  ],
250
  extensions=[
251
  ],
252
  nested_types=[],
253
  enum_types=[
254
  ],
255
  serialized_options=None,
256
  is_extendable=False,
257
  syntax='proto2',
258
  extension_ranges=[],
259
  oneofs=[
260
    _descriptor.OneofDescriptor(
261
      name='task_type', full_name='automl_zero.TaskSpec.task_type',
262
      index=0, containing_type=None, fields=[]),
263
  ],
264
  serialized_start=84,
265
  serialized_end=891,
266
)
267

268

269
_SCALARLINEARREGRESSIONTASKSPEC = _descriptor.Descriptor(
270
  name='ScalarLinearRegressionTaskSpec',
271
  full_name='automl_zero.ScalarLinearRegressionTaskSpec',
272
  filename=None,
273
  file=DESCRIPTOR,
274
  containing_type=None,
275
  fields=[
276
  ],
277
  extensions=[
278
  ],
279
  nested_types=[],
280
  enum_types=[
281
  ],
282
  serialized_options=None,
283
  is_extendable=False,
284
  syntax='proto2',
285
  extension_ranges=[],
286
  oneofs=[
287
  ],
288
  serialized_start=893,
289
  serialized_end=925,
290
)
291

292

293
_SCALAR2LAYERNNREGRESSIONTASKSPEC = _descriptor.Descriptor(
294
  name='Scalar2LayerNNRegressionTaskSpec',
295
  full_name='automl_zero.Scalar2LayerNNRegressionTaskSpec',
296
  filename=None,
297
  file=DESCRIPTOR,
298
  containing_type=None,
299
  fields=[
300
  ],
301
  extensions=[
302
  ],
303
  nested_types=[],
304
  enum_types=[
305
  ],
306
  serialized_options=None,
307
  is_extendable=False,
308
  syntax='proto2',
309
  extension_ranges=[],
310
  oneofs=[
311
  ],
312
  serialized_start=927,
313
  serialized_end=961,
314
)
315

316

317
_PROJECTEDBINARYCLASSIFICATIONTASK = _descriptor.Descriptor(
318
  name='ProjectedBinaryClassificationTask',
319
  full_name='automl_zero.ProjectedBinaryClassificationTask',
320
  filename=None,
321
  file=DESCRIPTOR,
322
  containing_type=None,
323
  fields=[
324
    _descriptor.FieldDescriptor(
325
      name='positive_class', full_name='automl_zero.ProjectedBinaryClassificationTask.positive_class', index=0,
326
      number=1, type=5, cpp_type=1, label=1,
327
      has_default_value=False, default_value=0,
328
      message_type=None, enum_type=None, containing_type=None,
329
      is_extension=False, extension_scope=None,
330
      serialized_options=None, file=DESCRIPTOR),
331
    _descriptor.FieldDescriptor(
332
      name='negative_class', full_name='automl_zero.ProjectedBinaryClassificationTask.negative_class', index=1,
333
      number=2, type=5, cpp_type=1, label=1,
334
      has_default_value=False, default_value=0,
335
      message_type=None, enum_type=None, containing_type=None,
336
      is_extension=False, extension_scope=None,
337
      serialized_options=None, file=DESCRIPTOR),
338
    _descriptor.FieldDescriptor(
339
      name='dataset_name', full_name='automl_zero.ProjectedBinaryClassificationTask.dataset_name', index=2,
340
      number=3, type=9, cpp_type=9, label=1,
341
      has_default_value=False, default_value=b"".decode('utf-8'),
342
      message_type=None, enum_type=None, containing_type=None,
343
      is_extension=False, extension_scope=None,
344
      serialized_options=None, file=DESCRIPTOR),
345
    _descriptor.FieldDescriptor(
346
      name='path', full_name='automl_zero.ProjectedBinaryClassificationTask.path', index=3,
347
      number=4, type=9, cpp_type=9, label=1,
348
      has_default_value=False, default_value=b"".decode('utf-8'),
349
      message_type=None, enum_type=None, containing_type=None,
350
      is_extension=False, extension_scope=None,
351
      serialized_options=None, file=DESCRIPTOR),
352
    _descriptor.FieldDescriptor(
353
      name='dataset', full_name='automl_zero.ProjectedBinaryClassificationTask.dataset', index=4,
354
      number=5, type=11, cpp_type=10, label=1,
355
      has_default_value=False, default_value=None,
356
      message_type=None, enum_type=None, containing_type=None,
357
      is_extension=False, extension_scope=None,
358
      serialized_options=None, file=DESCRIPTOR),
359
    _descriptor.FieldDescriptor(
360
      name='held_out_pairs', full_name='automl_zero.ProjectedBinaryClassificationTask.held_out_pairs', index=5,
361
      number=6, type=11, cpp_type=10, label=3,
362
      has_default_value=False, default_value=[],
363
      message_type=None, enum_type=None, containing_type=None,
364
      is_extension=False, extension_scope=None,
365
      serialized_options=None, file=DESCRIPTOR),
366
    _descriptor.FieldDescriptor(
367
      name='min_supported_data_seed', full_name='automl_zero.ProjectedBinaryClassificationTask.min_supported_data_seed', index=6,
368
      number=7, type=5, cpp_type=1, label=1,
369
      has_default_value=True, default_value=0,
370
      message_type=None, enum_type=None, containing_type=None,
371
      is_extension=False, extension_scope=None,
372
      serialized_options=None, file=DESCRIPTOR),
373
    _descriptor.FieldDescriptor(
374
      name='max_supported_data_seed', full_name='automl_zero.ProjectedBinaryClassificationTask.max_supported_data_seed', index=7,
375
      number=8, type=5, cpp_type=1, label=1,
376
      has_default_value=True, default_value=10,
377
      message_type=None, enum_type=None, containing_type=None,
378
      is_extension=False, extension_scope=None,
379
      serialized_options=None, file=DESCRIPTOR),
380
  ],
381
  extensions=[
382
  ],
383
  nested_types=[],
384
  enum_types=[
385
  ],
386
  serialized_options=None,
387
  is_extendable=False,
388
  syntax='proto2',
389
  extension_ranges=[],
390
  oneofs=[
391
    _descriptor.OneofDescriptor(
392
      name='task_source', full_name='automl_zero.ProjectedBinaryClassificationTask.task_source',
393
      index=0, containing_type=None, fields=[]),
394
  ],
395
  serialized_start=964,
396
  serialized_end=1273,
397
)
398

399

400
_CLASSPAIR = _descriptor.Descriptor(
401
  name='ClassPair',
402
  full_name='automl_zero.ClassPair',
403
  filename=None,
404
  file=DESCRIPTOR,
405
  containing_type=None,
406
  fields=[
407
    _descriptor.FieldDescriptor(
408
      name='positive_class', full_name='automl_zero.ClassPair.positive_class', index=0,
409
      number=1, type=5, cpp_type=1, label=1,
410
      has_default_value=False, default_value=0,
411
      message_type=None, enum_type=None, containing_type=None,
412
      is_extension=False, extension_scope=None,
413
      serialized_options=None, file=DESCRIPTOR),
414
    _descriptor.FieldDescriptor(
415
      name='negative_class', full_name='automl_zero.ClassPair.negative_class', index=1,
416
      number=2, type=5, cpp_type=1, label=1,
417
      has_default_value=False, default_value=0,
418
      message_type=None, enum_type=None, containing_type=None,
419
      is_extension=False, extension_scope=None,
420
      serialized_options=None, file=DESCRIPTOR),
421
  ],
422
  extensions=[
423
  ],
424
  nested_types=[],
425
  enum_types=[
426
  ],
427
  serialized_options=None,
428
  is_extendable=False,
429
  syntax='proto2',
430
  extension_ranges=[],
431
  oneofs=[
432
  ],
433
  serialized_start=1275,
434
  serialized_end=1334,
435
)
436

437

438
_SCALARLABELDATASET = _descriptor.Descriptor(
439
  name='ScalarLabelDataset',
440
  full_name='automl_zero.ScalarLabelDataset',
441
  filename=None,
442
  file=DESCRIPTOR,
443
  containing_type=None,
444
  fields=[
445
    _descriptor.FieldDescriptor(
446
      name='train_features', full_name='automl_zero.ScalarLabelDataset.train_features', index=0,
447
      number=1, type=11, cpp_type=10, label=3,
448
      has_default_value=False, default_value=[],
449
      message_type=None, enum_type=None, containing_type=None,
450
      is_extension=False, extension_scope=None,
451
      serialized_options=None, file=DESCRIPTOR),
452
    _descriptor.FieldDescriptor(
453
      name='train_labels', full_name='automl_zero.ScalarLabelDataset.train_labels', index=1,
454
      number=2, type=2, cpp_type=6, label=3,
455
      has_default_value=False, default_value=[],
456
      message_type=None, enum_type=None, containing_type=None,
457
      is_extension=False, extension_scope=None,
458
      serialized_options=None, file=DESCRIPTOR),
459
    _descriptor.FieldDescriptor(
460
      name='valid_features', full_name='automl_zero.ScalarLabelDataset.valid_features', index=2,
461
      number=3, type=11, cpp_type=10, label=3,
462
      has_default_value=False, default_value=[],
463
      message_type=None, enum_type=None, containing_type=None,
464
      is_extension=False, extension_scope=None,
465
      serialized_options=None, file=DESCRIPTOR),
466
    _descriptor.FieldDescriptor(
467
      name='valid_labels', full_name='automl_zero.ScalarLabelDataset.valid_labels', index=3,
468
      number=4, type=2, cpp_type=6, label=3,
469
      has_default_value=False, default_value=[],
470
      message_type=None, enum_type=None, containing_type=None,
471
      is_extension=False, extension_scope=None,
472
      serialized_options=None, file=DESCRIPTOR),
473
    _descriptor.FieldDescriptor(
474
      name='test_features', full_name='automl_zero.ScalarLabelDataset.test_features', index=4,
475
      number=5, type=11, cpp_type=10, label=3,
476
      has_default_value=False, default_value=[],
477
      message_type=None, enum_type=None, containing_type=None,
478
      is_extension=False, extension_scope=None,
479
      serialized_options=None, file=DESCRIPTOR),
480
    _descriptor.FieldDescriptor(
481
      name='test_labels', full_name='automl_zero.ScalarLabelDataset.test_labels', index=5,
482
      number=6, type=2, cpp_type=6, label=3,
483
      has_default_value=False, default_value=[],
484
      message_type=None, enum_type=None, containing_type=None,
485
      is_extension=False, extension_scope=None,
486
      serialized_options=None, file=DESCRIPTOR),
487
  ],
488
  extensions=[
489
  ],
490
  nested_types=[],
491
  enum_types=[
492
  ],
493
  serialized_options=None,
494
  is_extendable=False,
495
  syntax='proto2',
496
  extension_ranges=[],
497
  oneofs=[
498
  ],
499
  serialized_start=1337,
500
  serialized_end=1577,
501
)
502

503

504
_FEATUREVECTOR = _descriptor.Descriptor(
505
  name='FeatureVector',
506
  full_name='automl_zero.FeatureVector',
507
  filename=None,
508
  file=DESCRIPTOR,
509
  containing_type=None,
510
  fields=[
511
    _descriptor.FieldDescriptor(
512
      name='features', full_name='automl_zero.FeatureVector.features', index=0,
513
      number=1, type=2, cpp_type=6, label=3,
514
      has_default_value=False, default_value=[],
515
      message_type=None, enum_type=None, containing_type=None,
516
      is_extension=False, extension_scope=None,
517
      serialized_options=None, file=DESCRIPTOR),
518
  ],
519
  extensions=[
520
  ],
521
  nested_types=[],
522
  enum_types=[
523
  ],
524
  serialized_options=None,
525
  is_extendable=False,
526
  syntax='proto2',
527
  extension_ranges=[],
528
  oneofs=[
529
  ],
530
  serialized_start=1579,
531
  serialized_end=1612,
532
)
533

534

535
_UNITTESTFIXEDTASK = _descriptor.Descriptor(
536
  name='UnitTestFixedTask',
537
  full_name='automl_zero.UnitTestFixedTask',
538
  filename=None,
539
  file=DESCRIPTOR,
540
  containing_type=None,
541
  fields=[
542
    _descriptor.FieldDescriptor(
543
      name='train_features', full_name='automl_zero.UnitTestFixedTask.train_features', index=0,
544
      number=1, type=11, cpp_type=10, label=3,
545
      has_default_value=False, default_value=[],
546
      message_type=None, enum_type=None, containing_type=None,
547
      is_extension=False, extension_scope=None,
548
      serialized_options=None, file=DESCRIPTOR),
549
    _descriptor.FieldDescriptor(
550
      name='train_labels', full_name='automl_zero.UnitTestFixedTask.train_labels', index=1,
551
      number=2, type=11, cpp_type=10, label=3,
552
      has_default_value=False, default_value=[],
553
      message_type=None, enum_type=None, containing_type=None,
554
      is_extension=False, extension_scope=None,
555
      serialized_options=None, file=DESCRIPTOR),
556
    _descriptor.FieldDescriptor(
557
      name='valid_features', full_name='automl_zero.UnitTestFixedTask.valid_features', index=2,
558
      number=3, type=11, cpp_type=10, label=3,
559
      has_default_value=False, default_value=[],
560
      message_type=None, enum_type=None, containing_type=None,
561
      is_extension=False, extension_scope=None,
562
      serialized_options=None, file=DESCRIPTOR),
563
    _descriptor.FieldDescriptor(
564
      name='valid_labels', full_name='automl_zero.UnitTestFixedTask.valid_labels', index=3,
565
      number=4, type=11, cpp_type=10, label=3,
566
      has_default_value=False, default_value=[],
567
      message_type=None, enum_type=None, containing_type=None,
568
      is_extension=False, extension_scope=None,
569
      serialized_options=None, file=DESCRIPTOR),
570
    _descriptor.FieldDescriptor(
571
      name='test_features', full_name='automl_zero.UnitTestFixedTask.test_features', index=4,
572
      number=5, type=11, cpp_type=10, label=3,
573
      has_default_value=False, default_value=[],
574
      message_type=None, enum_type=None, containing_type=None,
575
      is_extension=False, extension_scope=None,
576
      serialized_options=None, file=DESCRIPTOR),
577
    _descriptor.FieldDescriptor(
578
      name='test_labels', full_name='automl_zero.UnitTestFixedTask.test_labels', index=5,
579
      number=6, type=11, cpp_type=10, label=3,
580
      has_default_value=False, default_value=[],
581
      message_type=None, enum_type=None, containing_type=None,
582
      is_extension=False, extension_scope=None,
583
      serialized_options=None, file=DESCRIPTOR),
584
  ],
585
  extensions=[
586
  ],
587
  nested_types=[],
588
  enum_types=[
589
  ],
590
  serialized_options=None,
591
  is_extendable=False,
592
  syntax='proto2',
593
  extension_ranges=[],
594
  oneofs=[
595
  ],
596
  serialized_start=1615,
597
  serialized_end=2004,
598
)
599

600

601
_UNITTESTFIXEDTASKVECTOR = _descriptor.Descriptor(
602
  name='UnitTestFixedTaskVector',
603
  full_name='automl_zero.UnitTestFixedTaskVector',
604
  filename=None,
605
  file=DESCRIPTOR,
606
  containing_type=None,
607
  fields=[
608
    _descriptor.FieldDescriptor(
609
      name='elements', full_name='automl_zero.UnitTestFixedTaskVector.elements', index=0,
610
      number=1, type=1, cpp_type=5, label=3,
611
      has_default_value=False, default_value=[],
612
      message_type=None, enum_type=None, containing_type=None,
613
      is_extension=False, extension_scope=None,
614
      serialized_options=None, file=DESCRIPTOR),
615
  ],
616
  extensions=[
617
  ],
618
  nested_types=[],
619
  enum_types=[
620
  ],
621
  serialized_options=None,
622
  is_extendable=False,
623
  syntax='proto2',
624
  extension_ranges=[],
625
  oneofs=[
626
  ],
627
  serialized_start=2006,
628
  serialized_end=2049,
629
)
630

631

632
_UNITTESTZEROSTASKSPEC = _descriptor.Descriptor(
633
  name='UnitTestZerosTaskSpec',
634
  full_name='automl_zero.UnitTestZerosTaskSpec',
635
  filename=None,
636
  file=DESCRIPTOR,
637
  containing_type=None,
638
  fields=[
639
  ],
640
  extensions=[
641
  ],
642
  nested_types=[],
643
  enum_types=[
644
  ],
645
  serialized_options=None,
646
  is_extendable=False,
647
  syntax='proto2',
648
  extension_ranges=[],
649
  oneofs=[
650
  ],
651
  serialized_start=2051,
652
  serialized_end=2074,
653
)
654

655

656
_UNITTESTONESTASKSPEC = _descriptor.Descriptor(
657
  name='UnitTestOnesTaskSpec',
658
  full_name='automl_zero.UnitTestOnesTaskSpec',
659
  filename=None,
660
  file=DESCRIPTOR,
661
  containing_type=None,
662
  fields=[
663
  ],
664
  extensions=[
665
  ],
666
  nested_types=[],
667
  enum_types=[
668
  ],
669
  serialized_options=None,
670
  is_extendable=False,
671
  syntax='proto2',
672
  extension_ranges=[],
673
  oneofs=[
674
  ],
675
  serialized_start=2076,
676
  serialized_end=2098,
677
)
678

679

680
_UNITTESTINCREMENTTASKSPEC = _descriptor.Descriptor(
681
  name='UnitTestIncrementTaskSpec',
682
  full_name='automl_zero.UnitTestIncrementTaskSpec',
683
  filename=None,
684
  file=DESCRIPTOR,
685
  containing_type=None,
686
  fields=[
687
    _descriptor.FieldDescriptor(
688
      name='increment', full_name='automl_zero.UnitTestIncrementTaskSpec.increment', index=0,
689
      number=1, type=1, cpp_type=5, label=1,
690
      has_default_value=True, default_value=float(1),
691
      message_type=None, enum_type=None, containing_type=None,
692
      is_extension=False, extension_scope=None,
693
      serialized_options=None, file=DESCRIPTOR),
694
  ],
695
  extensions=[
696
  ],
697
  nested_types=[],
698
  enum_types=[
699
  ],
700
  serialized_options=None,
701
  is_extendable=False,
702
  syntax='proto2',
703
  extension_ranges=[],
704
  oneofs=[
705
  ],
706
  serialized_start=2100,
707
  serialized_end=2149,
708
)
709

710
_TASKCOLLECTION.fields_by_name['tasks'].message_type = _TASKSPEC
711
_TASKSPEC.fields_by_name['eval_type'].enum_type = _EVALTYPE
712
_TASKSPEC.fields_by_name['scalar_linear_regression_task'].message_type = _SCALARLINEARREGRESSIONTASKSPEC
713
_TASKSPEC.fields_by_name['scalar_2layer_nn_regression_task'].message_type = _SCALAR2LAYERNNREGRESSIONTASKSPEC
714
_TASKSPEC.fields_by_name['projected_binary_classification_task'].message_type = _PROJECTEDBINARYCLASSIFICATIONTASK
715
_TASKSPEC.fields_by_name['unit_test_fixed_task'].message_type = _UNITTESTFIXEDTASK
716
_TASKSPEC.fields_by_name['unit_test_zeros_task'].message_type = _UNITTESTZEROSTASKSPEC
717
_TASKSPEC.fields_by_name['unit_test_ones_task'].message_type = _UNITTESTONESTASKSPEC
718
_TASKSPEC.fields_by_name['unit_test_increment_task'].message_type = _UNITTESTINCREMENTTASKSPEC
719
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
720
  _TASKSPEC.fields_by_name['scalar_linear_regression_task'])
721
_TASKSPEC.fields_by_name['scalar_linear_regression_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
722
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
723
  _TASKSPEC.fields_by_name['scalar_2layer_nn_regression_task'])
724
_TASKSPEC.fields_by_name['scalar_2layer_nn_regression_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
725
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
726
  _TASKSPEC.fields_by_name['projected_binary_classification_task'])
727
_TASKSPEC.fields_by_name['projected_binary_classification_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
728
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
729
  _TASKSPEC.fields_by_name['unit_test_fixed_task'])
730
_TASKSPEC.fields_by_name['unit_test_fixed_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
731
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
732
  _TASKSPEC.fields_by_name['unit_test_zeros_task'])
733
_TASKSPEC.fields_by_name['unit_test_zeros_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
734
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
735
  _TASKSPEC.fields_by_name['unit_test_ones_task'])
736
_TASKSPEC.fields_by_name['unit_test_ones_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
737
_TASKSPEC.oneofs_by_name['task_type'].fields.append(
738
  _TASKSPEC.fields_by_name['unit_test_increment_task'])
739
_TASKSPEC.fields_by_name['unit_test_increment_task'].containing_oneof = _TASKSPEC.oneofs_by_name['task_type']
740
_PROJECTEDBINARYCLASSIFICATIONTASK.fields_by_name['dataset'].message_type = _SCALARLABELDATASET
741
_PROJECTEDBINARYCLASSIFICATIONTASK.fields_by_name['held_out_pairs'].message_type = _CLASSPAIR
742
_PROJECTEDBINARYCLASSIFICATIONTASK.oneofs_by_name['task_source'].fields.append(
743
  _PROJECTEDBINARYCLASSIFICATIONTASK.fields_by_name['path'])
744
_PROJECTEDBINARYCLASSIFICATIONTASK.fields_by_name['path'].containing_oneof = _PROJECTEDBINARYCLASSIFICATIONTASK.oneofs_by_name['task_source']
745
_PROJECTEDBINARYCLASSIFICATIONTASK.oneofs_by_name['task_source'].fields.append(
746
  _PROJECTEDBINARYCLASSIFICATIONTASK.fields_by_name['dataset'])
747
_PROJECTEDBINARYCLASSIFICATIONTASK.fields_by_name['dataset'].containing_oneof = _PROJECTEDBINARYCLASSIFICATIONTASK.oneofs_by_name['task_source']
748
_SCALARLABELDATASET.fields_by_name['train_features'].message_type = _FEATUREVECTOR
749
_SCALARLABELDATASET.fields_by_name['valid_features'].message_type = _FEATUREVECTOR
750
_SCALARLABELDATASET.fields_by_name['test_features'].message_type = _FEATUREVECTOR
751
_UNITTESTFIXEDTASK.fields_by_name['train_features'].message_type = _UNITTESTFIXEDTASKVECTOR
752
_UNITTESTFIXEDTASK.fields_by_name['train_labels'].message_type = _UNITTESTFIXEDTASKVECTOR
753
_UNITTESTFIXEDTASK.fields_by_name['valid_features'].message_type = _UNITTESTFIXEDTASKVECTOR
754
_UNITTESTFIXEDTASK.fields_by_name['valid_labels'].message_type = _UNITTESTFIXEDTASKVECTOR
755
_UNITTESTFIXEDTASK.fields_by_name['test_features'].message_type = _UNITTESTFIXEDTASKVECTOR
756
_UNITTESTFIXEDTASK.fields_by_name['test_labels'].message_type = _UNITTESTFIXEDTASKVECTOR
757
DESCRIPTOR.message_types_by_name['TaskCollection'] = _TASKCOLLECTION
758
DESCRIPTOR.message_types_by_name['TaskSpec'] = _TASKSPEC
759
DESCRIPTOR.message_types_by_name['ScalarLinearRegressionTaskSpec'] = _SCALARLINEARREGRESSIONTASKSPEC
760
DESCRIPTOR.message_types_by_name['Scalar2LayerNNRegressionTaskSpec'] = _SCALAR2LAYERNNREGRESSIONTASKSPEC
761
DESCRIPTOR.message_types_by_name['ProjectedBinaryClassificationTask'] = _PROJECTEDBINARYCLASSIFICATIONTASK
762
DESCRIPTOR.message_types_by_name['ClassPair'] = _CLASSPAIR
763
DESCRIPTOR.message_types_by_name['ScalarLabelDataset'] = _SCALARLABELDATASET
764
DESCRIPTOR.message_types_by_name['FeatureVector'] = _FEATUREVECTOR
765
DESCRIPTOR.message_types_by_name['UnitTestFixedTask'] = _UNITTESTFIXEDTASK
766
DESCRIPTOR.message_types_by_name['UnitTestFixedTaskVector'] = _UNITTESTFIXEDTASKVECTOR
767
DESCRIPTOR.message_types_by_name['UnitTestZerosTaskSpec'] = _UNITTESTZEROSTASKSPEC
768
DESCRIPTOR.message_types_by_name['UnitTestOnesTaskSpec'] = _UNITTESTONESTASKSPEC
769
DESCRIPTOR.message_types_by_name['UnitTestIncrementTaskSpec'] = _UNITTESTINCREMENTTASKSPEC
770
DESCRIPTOR.enum_types_by_name['EvalType'] = _EVALTYPE
771
DESCRIPTOR.enum_types_by_name['ActivationType'] = _ACTIVATIONTYPE
772
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
773

774
TaskCollection = _reflection.GeneratedProtocolMessageType('TaskCollection', (_message.Message,), {
775
  'DESCRIPTOR' : _TASKCOLLECTION,
776
  '__module__' : 'task_pb2'
777
  # @@protoc_insertion_point(class_scope:automl_zero.TaskCollection)
778
  })
779
_sym_db.RegisterMessage(TaskCollection)
780

781
TaskSpec = _reflection.GeneratedProtocolMessageType('TaskSpec', (_message.Message,), {
782
  'DESCRIPTOR' : _TASKSPEC,
783
  '__module__' : 'task_pb2'
784
  # @@protoc_insertion_point(class_scope:automl_zero.TaskSpec)
785
  })
786
_sym_db.RegisterMessage(TaskSpec)
787

788
ScalarLinearRegressionTaskSpec = _reflection.GeneratedProtocolMessageType('ScalarLinearRegressionTaskSpec', (_message.Message,), {
789
  'DESCRIPTOR' : _SCALARLINEARREGRESSIONTASKSPEC,
790
  '__module__' : 'task_pb2'
791
  # @@protoc_insertion_point(class_scope:automl_zero.ScalarLinearRegressionTaskSpec)
792
  })
793
_sym_db.RegisterMessage(ScalarLinearRegressionTaskSpec)
794

795
Scalar2LayerNNRegressionTaskSpec = _reflection.GeneratedProtocolMessageType('Scalar2LayerNNRegressionTaskSpec', (_message.Message,), {
796
  'DESCRIPTOR' : _SCALAR2LAYERNNREGRESSIONTASKSPEC,
797
  '__module__' : 'task_pb2'
798
  # @@protoc_insertion_point(class_scope:automl_zero.Scalar2LayerNNRegressionTaskSpec)
799
  })
800
_sym_db.RegisterMessage(Scalar2LayerNNRegressionTaskSpec)
801

802
ProjectedBinaryClassificationTask = _reflection.GeneratedProtocolMessageType('ProjectedBinaryClassificationTask', (_message.Message,), {
803
  'DESCRIPTOR' : _PROJECTEDBINARYCLASSIFICATIONTASK,
804
  '__module__' : 'task_pb2'
805
  # @@protoc_insertion_point(class_scope:automl_zero.ProjectedBinaryClassificationTask)
806
  })
807
_sym_db.RegisterMessage(ProjectedBinaryClassificationTask)
808

809
ClassPair = _reflection.GeneratedProtocolMessageType('ClassPair', (_message.Message,), {
810
  'DESCRIPTOR' : _CLASSPAIR,
811
  '__module__' : 'task_pb2'
812
  # @@protoc_insertion_point(class_scope:automl_zero.ClassPair)
813
  })
814
_sym_db.RegisterMessage(ClassPair)
815

816
ScalarLabelDataset = _reflection.GeneratedProtocolMessageType('ScalarLabelDataset', (_message.Message,), {
817
  'DESCRIPTOR' : _SCALARLABELDATASET,
818
  '__module__' : 'task_pb2'
819
  # @@protoc_insertion_point(class_scope:automl_zero.ScalarLabelDataset)
820
  })
821
_sym_db.RegisterMessage(ScalarLabelDataset)
822

823
FeatureVector = _reflection.GeneratedProtocolMessageType('FeatureVector', (_message.Message,), {
824
  'DESCRIPTOR' : _FEATUREVECTOR,
825
  '__module__' : 'task_pb2'
826
  # @@protoc_insertion_point(class_scope:automl_zero.FeatureVector)
827
  })
828
_sym_db.RegisterMessage(FeatureVector)
829

830
UnitTestFixedTask = _reflection.GeneratedProtocolMessageType('UnitTestFixedTask', (_message.Message,), {
831
  'DESCRIPTOR' : _UNITTESTFIXEDTASK,
832
  '__module__' : 'task_pb2'
833
  # @@protoc_insertion_point(class_scope:automl_zero.UnitTestFixedTask)
834
  })
835
_sym_db.RegisterMessage(UnitTestFixedTask)
836

837
UnitTestFixedTaskVector = _reflection.GeneratedProtocolMessageType('UnitTestFixedTaskVector', (_message.Message,), {
838
  'DESCRIPTOR' : _UNITTESTFIXEDTASKVECTOR,
839
  '__module__' : 'task_pb2'
840
  # @@protoc_insertion_point(class_scope:automl_zero.UnitTestFixedTaskVector)
841
  })
842
_sym_db.RegisterMessage(UnitTestFixedTaskVector)
843

844
UnitTestZerosTaskSpec = _reflection.GeneratedProtocolMessageType('UnitTestZerosTaskSpec', (_message.Message,), {
845
  'DESCRIPTOR' : _UNITTESTZEROSTASKSPEC,
846
  '__module__' : 'task_pb2'
847
  # @@protoc_insertion_point(class_scope:automl_zero.UnitTestZerosTaskSpec)
848
  })
849
_sym_db.RegisterMessage(UnitTestZerosTaskSpec)
850

851
UnitTestOnesTaskSpec = _reflection.GeneratedProtocolMessageType('UnitTestOnesTaskSpec', (_message.Message,), {
852
  'DESCRIPTOR' : _UNITTESTONESTASKSPEC,
853
  '__module__' : 'task_pb2'
854
  # @@protoc_insertion_point(class_scope:automl_zero.UnitTestOnesTaskSpec)
855
  })
856
_sym_db.RegisterMessage(UnitTestOnesTaskSpec)
857

858
UnitTestIncrementTaskSpec = _reflection.GeneratedProtocolMessageType('UnitTestIncrementTaskSpec', (_message.Message,), {
859
  'DESCRIPTOR' : _UNITTESTINCREMENTTASKSPEC,
860
  '__module__' : 'task_pb2'
861
  # @@protoc_insertion_point(class_scope:automl_zero.UnitTestIncrementTaskSpec)
862
  })
863
_sym_db.RegisterMessage(UnitTestIncrementTaskSpec)
864

865

866
# @@protoc_insertion_point(module_scope)
867