@@ -1167,16 +1167,13 @@ def run_post_training_bias(
1167
1167
the Trial Component will be unassociated.
1168
1168
* ``'TrialComponentDisplayName'`` is used for display in Amazon SageMaker Studio.
1169
1169
""" # noqa E501 # pylint: disable=c0301
1170
- analysis_config = data_config .get_config ()
1171
- analysis_config .update (data_bias_config .get_config ())
1172
- (
1173
- probability_threshold ,
1174
- predictor_config ,
1175
- ) = model_predicted_label_config .get_predictor_config ()
1176
- predictor_config .update (model_config .get_predictor_config ())
1177
- analysis_config ["methods" ] = {"post_training_bias" : {"methods" : methods }}
1178
- analysis_config ["predictor" ] = predictor_config
1179
- _set (probability_threshold , "probability_threshold" , analysis_config )
1170
+ analysis_config = _AnalysisConfigGenerator .bias_post_training (
1171
+ data_config ,
1172
+ data_bias_config ,
1173
+ model_predicted_label_config ,
1174
+ methods ,
1175
+ model_config
1176
+ )
1180
1177
if job_name is None :
1181
1178
if self .job_name_prefix :
1182
1179
job_name = utils .name_from_base (self .job_name_prefix )
@@ -1447,6 +1444,26 @@ def bias_pre_training(data_config, data_bias_config, methods):
1447
1444
analysis_config ["methods" ] = {"pre_training_bias" : {"methods" : methods }}
1448
1445
return analysis_config
1449
1446
1447
+ @staticmethod
1448
+ def bias_post_training (
1449
+ data_config ,
1450
+ data_bias_config ,
1451
+ model_predicted_label_config ,
1452
+ methods ,
1453
+ model_config
1454
+ ):
1455
+ analysis_config = data_config .get_config ()
1456
+ analysis_config .update (data_bias_config .get_config ())
1457
+ analysis_config ["methods" ] = {"post_training_bias" : {"methods" : methods }}
1458
+ (
1459
+ probability_threshold ,
1460
+ predictor_config ,
1461
+ ) = model_predicted_label_config .get_predictor_config ()
1462
+ predictor_config .update (model_config .get_predictor_config ())
1463
+ analysis_config ["predictor" ] = predictor_config
1464
+ _set (probability_threshold , "probability_threshold" , analysis_config )
1465
+ return analysis_config
1466
+
1450
1467
@staticmethod
1451
1468
def _common (analysis_config ):
1452
1469
analysis_config ["methods" ]["report" ] = {
0 commit comments