@@ -33,14 +33,14 @@ class DataSplitMethod(enum.IntEnum):
3333 AUTO_SPLIT (int): Splits data automatically: Uses NO\_SPLIT if the data size is small.
3434 Otherwise uses RANDOM.
3535 """
36-
3736 DATA_SPLIT_METHOD_UNSPECIFIED = 0
3837 RANDOM = 1
3938 CUSTOM = 2
4039 SEQUENTIAL = 3
4140 NO_SPLIT = 4
4241 AUTO_SPLIT = 5
4342
43+
4444 class DistanceType (enum .IntEnum ):
4545 """
4646 Distance metric used to compute the distance between two points.
@@ -50,11 +50,11 @@ class DistanceType(enum.IntEnum):
5050 EUCLIDEAN (int): Eculidean distance.
5151 COSINE (int): Cosine distance.
5252 """
53-
5453 DISTANCE_TYPE_UNSPECIFIED = 0
5554 EUCLIDEAN = 1
5655 COSINE = 2
5756
57+
5858 class LearnRateStrategy (enum .IntEnum ):
5959 """
6060 Indicates the learning rate optimization strategy to use.
@@ -64,11 +64,11 @@ class LearnRateStrategy(enum.IntEnum):
6464 LINE_SEARCH (int): Use line search to determine learning rate.
6565 CONSTANT (int): Use a constant learning rate.
6666 """
67-
6867 LEARN_RATE_STRATEGY_UNSPECIFIED = 0
6968 LINE_SEARCH = 1
7069 CONSTANT = 2
7170
71+
7272 class LossType (enum .IntEnum ):
7373 """
7474 Loss metric to evaluate model training performance.
@@ -78,11 +78,11 @@ class LossType(enum.IntEnum):
7878 MEAN_SQUARED_LOSS (int): Mean squared loss, used for linear regression.
7979 MEAN_LOG_LOSS (int): Mean log loss, used for logistic regression.
8080 """
81-
8281 LOSS_TYPE_UNSPECIFIED = 0
8382 MEAN_SQUARED_LOSS = 1
8483 MEAN_LOG_LOSS = 2
8584
85+
8686 class ModelType (enum .IntEnum ):
8787 """
8888 Indicates the type of the Model.
@@ -94,13 +94,13 @@ class ModelType(enum.IntEnum):
9494 KMEANS (int): K-means clustering model.
9595 TENSORFLOW (int): [Beta] An imported TensorFlow model.
9696 """
97-
9897 MODEL_TYPE_UNSPECIFIED = 0
9998 LINEAR_REGRESSION = 1
10099 LOGISTIC_REGRESSION = 2
101100 KMEANS = 3
102101 TENSORFLOW = 6
103102
103+
104104 class OptimizationStrategy (enum .IntEnum ):
105105 """
106106 Indicates the optimization strategy used for training.
@@ -110,11 +110,11 @@ class OptimizationStrategy(enum.IntEnum):
110110 BATCH_GRADIENT_DESCENT (int): Uses an iterative batch gradient descent algorithm.
111111 NORMAL_EQUATION (int): Uses a normal equation to solve linear regression problem.
112112 """
113-
114113 OPTIMIZATION_STRATEGY_UNSPECIFIED = 0
115114 BATCH_GRADIENT_DESCENT = 1
116115 NORMAL_EQUATION = 2
117116
117+
118118 class KmeansEnums (object ):
119119 class KmeansInitializationMethod (enum .IntEnum ):
120120 """
@@ -127,7 +127,6 @@ class KmeansInitializationMethod(enum.IntEnum):
127127 CUSTOM (int): Initializes the centroids using data specified in
128128 kmeans\_initialization\_column.
129129 """
130-
131130 KMEANS_INITIALIZATION_METHOD_UNSPECIFIED = 0
132131 RANDOM = 1
133132 CUSTOM = 2
@@ -154,7 +153,6 @@ class TypeKind(enum.IntEnum):
154153 STRUCT (int): Encoded as a list with fields of type Type.struct\_type[i]. List is used
155154 because a JSON object cannot have duplicate field names.
156155 """
157-
158156 TYPE_KIND_UNSPECIFIED = 0
159157 INT64 = 2
160158 BOOL = 5
@@ -169,3 +167,4 @@ class TypeKind(enum.IntEnum):
169167 NUMERIC = 23
170168 ARRAY = 16
171169 STRUCT = 17
170+
0 commit comments