Skip to content

Commit 57b2b2b

Browse files
committed
update integration test
1 parent 3073436 commit 57b2b2b

File tree

1 file changed

+19
-19
lines changed

1 file changed

+19
-19
lines changed

tests/integration/single_table/data_augmentation/test_binary_classifier_recall_efficacy.py

+19-19
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,13 @@ def test_end_to_end(self):
5555
},
5656
'augmented_data': {
5757
'precision_score_training': 0.8034682080924855,
58-
'recall_score_validation': 0.8461538461538461,
59-
'precision_score_validation': 0.44,
58+
'recall_score_validation': 0.7692307692307693,
59+
'precision_score_validation': 0.4,
6060
'prediction_counts_validation': {
61-
'true_positive': 11,
62-
'false_positive': 14,
63-
'true_negative': 11,
64-
'false_negative': 2,
61+
'true_positive': 10,
62+
'false_positive': 15,
63+
'true_negative': 10,
64+
'false_negative': 3,
6565
},
6666
},
6767
'parameters': {
@@ -136,23 +136,23 @@ def test_with_nan_target_column(self):
136136
expected_result = {
137137
'real_data_baseline': {
138138
'precision_score_training': 0.8082191780821918,
139-
'recall_score_validation': 0.6153846153846154,
140-
'precision_score_validation': 0.38095238095238093,
139+
'recall_score_validation': 0.6923076923076923,
140+
'precision_score_validation': 0.391304347826087,
141141
'prediction_counts_validation': {
142-
'true_positive': 8,
143-
'false_positive': 13,
144-
'true_negative': 20,
145-
'false_negative': 5,
142+
'true_positive': 9,
143+
'false_positive': 14,
144+
'true_negative': 19,
145+
'false_negative': 4,
146146
},
147147
},
148148
'augmented_data': {
149149
'precision_score_training': 0.8035714285714286,
150150
'recall_score_validation': 0.7692307692307693,
151-
'precision_score_validation': 0.4,
151+
'precision_score_validation': 0.38461538461538464,
152152
'prediction_counts_validation': {
153153
'true_positive': 10,
154-
'false_positive': 15,
155-
'true_negative': 18,
154+
'false_positive': 16,
155+
'true_negative': 17,
156156
'false_negative': 3,
157157
},
158158
},
@@ -162,7 +162,7 @@ def test_with_nan_target_column(self):
162162
'classifier': 'XGBoost',
163163
'fixed_recall_value': 0.8,
164164
},
165-
'score': 0.15384615384615385,
165+
'score': 0.07692307692307698,
166166
}
167167
assert result_breakdown == expected_result
168168

@@ -230,11 +230,11 @@ def test_with_multi_class(self):
230230
'augmented_data': {
231231
'precision_score_training': 0.8,
232232
'recall_score_validation': 1.0,
233-
'precision_score_validation': 0.4642857142857143,
233+
'precision_score_validation': 0.4482758620689655,
234234
'prediction_counts_validation': {
235235
'true_positive': 13,
236-
'false_positive': 15,
237-
'true_negative': 10,
236+
'false_positive': 16,
237+
'true_negative': 9,
238238
'false_negative': 0,
239239
},
240240
},

0 commit comments

Comments
 (0)