Skip to content

Commit 013ba5f

Browse files
committed
fix logisticregression bug in python
1 parent b9c3a51 commit 013ba5f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

flink-ml-python/pyflink/ml/classification/tests/test_logisticregression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def test_save_load_and_predict(self):
120120
model = regression.fit(self.binomial_data_table)
121121
self.assertEqual(
122122
model.get_model_data()[0].get_schema().get_field_names(),
123-
['coefficient', 'modelVersion'])
123+
['coefficient', "startIndex", "endIndex", 'modelVersion'])
124124
output = model.transform(self.binomial_data_table)[0]
125125
field_names = output.get_schema().get_field_names()
126126
self.verify_predict_result(

0 commit comments

Comments
 (0)