diff --git a/src/triage/component/postmodeling/contrast/model_group_evaluator.py b/src/triage/component/postmodeling/contrast/model_group_evaluator.py index 50d141f03..8af111828 100644 --- a/src/triage/component/postmodeling/contrast/model_group_evaluator.py +++ b/src/triage/component/postmodeling/contrast/model_group_evaluator.py @@ -272,8 +272,8 @@ def plot_prec_across_time(self, model_metrics = self.metrics model_metrics[['param', 'param_type']] = \ model_metrics['parameter'].str.split('_', 1, expand=True) - model_metrics['param'] = model_metrics['param'].astype(str).astype(float) - model_metrics['param_type'] = model_metrics['param_type'].apply(lambda x: 'rank_'+x) + model_metrics['param'] = model_metrics['param'].replace('',np.nan).astype('float') + model_metrics['param_type'] = model_metrics['param_type'].apply(lambda x: f'rank_{x}' if x is not None else x) # Filter model_group_id metrics and create pivot table by each # model_group_id.