@@ -55,13 +55,13 @@ def test_end_to_end(self):
55
55
},
56
56
'augmented_data' : {
57
57
'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 ,
60
60
'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 ,
65
65
},
66
66
},
67
67
'parameters' : {
@@ -136,23 +136,23 @@ def test_with_nan_target_column(self):
136
136
expected_result = {
137
137
'real_data_baseline' : {
138
138
'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 ,
141
141
'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 ,
146
146
},
147
147
},
148
148
'augmented_data' : {
149
149
'precision_score_training' : 0.8035714285714286 ,
150
150
'recall_score_validation' : 0.7692307692307693 ,
151
- 'precision_score_validation' : 0.4 ,
151
+ 'precision_score_validation' : 0.38461538461538464 ,
152
152
'prediction_counts_validation' : {
153
153
'true_positive' : 10 ,
154
- 'false_positive' : 15 ,
155
- 'true_negative' : 18 ,
154
+ 'false_positive' : 16 ,
155
+ 'true_negative' : 17 ,
156
156
'false_negative' : 3 ,
157
157
},
158
158
},
@@ -162,7 +162,7 @@ def test_with_nan_target_column(self):
162
162
'classifier' : 'XGBoost' ,
163
163
'fixed_recall_value' : 0.8 ,
164
164
},
165
- 'score' : 0.15384615384615385 ,
165
+ 'score' : 0.07692307692307698 ,
166
166
}
167
167
assert result_breakdown == expected_result
168
168
@@ -230,11 +230,11 @@ def test_with_multi_class(self):
230
230
'augmented_data' : {
231
231
'precision_score_training' : 0.8 ,
232
232
'recall_score_validation' : 1.0 ,
233
- 'precision_score_validation' : 0.4642857142857143 ,
233
+ 'precision_score_validation' : 0.4482758620689655 ,
234
234
'prediction_counts_validation' : {
235
235
'true_positive' : 13 ,
236
- 'false_positive' : 15 ,
237
- 'true_negative' : 10 ,
236
+ 'false_positive' : 16 ,
237
+ 'true_negative' : 9 ,
238
238
'false_negative' : 0 ,
239
239
},
240
240
},
0 commit comments