@@ -85,6 +85,17 @@ def mock_X(mock_y, mock_session):
85
85
["index_column_id" ],
86
86
["index_column_label" ],
87
87
)
88
+
89
+ mock_X .join (mock_y ).reset_index (drop = True ).sql = "input_X_y_no_index_sql"
90
+ mock_X .join (mock_y ).reset_index (drop = True ).cache .return_value = mock_X .join (
91
+ mock_y
92
+ ).reset_index (drop = True )
93
+ mock_X .join (mock_y ).reset_index (drop = True )._to_sql_query .return_value = (
94
+ "input_X_y_no_index_sql" ,
95
+ ["index_column_id" ],
96
+ ["index_column_label" ],
97
+ )
98
+
88
99
mock_X .cache .return_value = mock_X
89
100
90
101
return mock_X
@@ -107,7 +118,7 @@ def test_linear_regression_default_fit(
107
118
model .fit (mock_X , mock_y )
108
119
109
120
mock_session ._start_query_ml_ddl .assert_called_once_with (
110
- "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LINEAR_REG',\n data_split_method='NO_SPLIT',\n optimize_strategy='auto_strategy',\n fit_intercept=True,\n l2_reg=0.0,\n max_iterations=20,\n learn_rate_strategy='line_search',\n min_rel_progress=0.01,\n calculate_p_values=False,\n enable_global_explain=False,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_sql "
121
+ "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LINEAR_REG',\n data_split_method='NO_SPLIT',\n optimize_strategy='auto_strategy',\n fit_intercept=True,\n l2_reg=0.0,\n max_iterations=20,\n learn_rate_strategy='line_search',\n min_rel_progress=0.01,\n calculate_p_values=False,\n enable_global_explain=False,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_no_index_sql "
111
122
)
112
123
113
124
@@ -117,7 +128,7 @@ def test_linear_regression_params_fit(bqml_model_factory, mock_session, mock_X,
117
128
model .fit (mock_X , mock_y )
118
129
119
130
mock_session ._start_query_ml_ddl .assert_called_once_with (
120
- "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LINEAR_REG',\n data_split_method='NO_SPLIT',\n optimize_strategy='auto_strategy',\n fit_intercept=False,\n l2_reg=0.0,\n max_iterations=20,\n learn_rate_strategy='line_search',\n min_rel_progress=0.01,\n calculate_p_values=False,\n enable_global_explain=False,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_sql "
131
+ "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LINEAR_REG',\n data_split_method='NO_SPLIT',\n optimize_strategy='auto_strategy',\n fit_intercept=False,\n l2_reg=0.0,\n max_iterations=20,\n learn_rate_strategy='line_search',\n min_rel_progress=0.01,\n calculate_p_values=False,\n enable_global_explain=False,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_no_index_sql "
121
132
)
122
133
123
134
@@ -150,7 +161,7 @@ def test_logistic_regression_default_fit(
150
161
model .fit (mock_X , mock_y )
151
162
152
163
mock_session ._start_query_ml_ddl .assert_called_once_with (
153
- "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LOGISTIC_REG',\n data_split_method='NO_SPLIT',\n fit_intercept=True,\n auto_class_weights=False,\n optimize_strategy='auto_strategy',\n l2_reg=0.0,\n max_iterations=20,\n learn_rate_strategy='line_search',\n min_rel_progress=0.01,\n calculate_p_values=False,\n enable_global_explain=False,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_sql "
164
+ "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LOGISTIC_REG',\n data_split_method='NO_SPLIT',\n fit_intercept=True,\n auto_class_weights=False,\n optimize_strategy='auto_strategy',\n l2_reg=0.0,\n max_iterations=20,\n learn_rate_strategy='line_search',\n min_rel_progress=0.01,\n calculate_p_values=False,\n enable_global_explain=False,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_no_index_sql "
154
165
)
155
166
156
167
@@ -172,7 +183,7 @@ def test_logistic_regression_params_fit(
172
183
model .fit (mock_X , mock_y )
173
184
174
185
mock_session ._start_query_ml_ddl .assert_called_once_with (
175
- "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LOGISTIC_REG',\n data_split_method='NO_SPLIT',\n fit_intercept=False,\n auto_class_weights=True,\n optimize_strategy='batch_gradient_descent',\n l2_reg=0.2,\n max_iterations=30,\n learn_rate_strategy='constant',\n min_rel_progress=0.02,\n calculate_p_values=False,\n enable_global_explain=False,\n l1_reg=0.2,\n learn_rate=0.2,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_sql "
186
+ "CREATE OR REPLACE MODEL `test-project`.`_anon123`.`temp_model_id`\n OPTIONS(\n model_type='LOGISTIC_REG',\n data_split_method='NO_SPLIT',\n fit_intercept=False,\n auto_class_weights=True,\n optimize_strategy='batch_gradient_descent',\n l2_reg=0.2,\n max_iterations=30,\n learn_rate_strategy='constant',\n min_rel_progress=0.02,\n calculate_p_values=False,\n enable_global_explain=False,\n l1_reg=0.2,\n learn_rate=0.2,\n INPUT_LABEL_COLS=['input_column_label'])\n AS input_X_y_no_index_sql "
176
187
)
177
188
178
189
0 commit comments