Fixes #691

dssg · May 9, 2019 · 759df4e · 759df4e
1 parent cc5f66c
commit 759df4e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.