Skip to content

Commit bfff6b5

Browse files
committed
Updates for scikit-learn 1.4
Signed-off-by: Beat Buesser <[email protected]>
1 parent 8068897 commit bfff6b5

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

tests/attacks/evasion/test_lowprofool.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@ def pearson_correlations(x, y):
329329
return result
330330

331331
# Setup classifier
332-
lr_clf = LogisticRegression(penalty=None)
332+
lr_clf = LogisticRegression(penalty="l2")
333333
lr_clf.fit(x_train, y_train)
334334
clf_slr = ScikitlearnLogisticRegression(model=lr_clf, clip_values=clip_values)
335335

@@ -392,7 +392,7 @@ def test_clipping(iris_dataset):
392392
(x_train, y_train, x_valid, y_valid), _, clip_values = iris_dataset
393393

394394
# Setup classifier
395-
lr_clf = LogisticRegression(penalty=None)
395+
lr_clf = LogisticRegression(penalty="l2")
396396
lr_clf.fit(x_train, y_train)
397397

398398
# Dataset min-max clipping values

tests/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1748,7 +1748,6 @@ def get_tabular_classifier_scikit_list(clipped=False, model_list_names=None):
17481748
)
17491749

17501750
sklearn_version = list(map(int, sklearn.__version__.split(".")))
1751-
sklearn_ge_1_3_0 = sklearn_version[0] == 1 and sklearn_version[1] == 3
17521751
if sklearn_version[0] == 1 and sklearn_version[1] == 3:
17531752
suffix = "-eq-1.3.0"
17541753
elif sklearn_version[0] == 1 and sklearn_version[1] == 4:

0 commit comments

Comments
 (0)